Ultralytics YOLO27:

Bộ dữ liệu COCO8-Grayscale#

Giới thiệu#

Bộ dữ liệu COCO8-Grayscale của Ultralytics là một bộ dữ liệu phát hiện đối tượng nhỏ gọn nhưng mạnh mẽ, gồm 8 hình ảnh đầu tiên từ tập COCO train 2017 và được chuyển đổi sang định dạng grayscale—4 ảnh dùng để huấn luyện và 4 ảnh dùng để validation. Bộ dữ liệu này được thiết kế chuyên biệt cho việc kiểm thử nhanh, debug và thử nghiệm với các model grayscale và pipeline huấn luyện YOLO. Kích thước nhỏ giúp bộ dữ liệu dễ quản lý, trong khi tính đa dạng bảo đảm đây là công cụ sanity check hiệu quả trước khi mở rộng lên các bộ dữ liệu lớn hơn.



Watch: How to Train Ultralytics YOLO on Grayscale and Multispectral Datasets | Multi-Channel VisionAI 🚀

Cấu trúc bộ dữ liệu#

COCO8-Grayscale mô phỏng COCO8—8 hình ảnh đầu tiên từ tập COCO train 2017, được chia thành 4 ảnh huấn luyện và 4 ảnh validation trên toàn bộ 80 class đối tượng COCO—nhưng mỗi ảnh được lưu dưới dạng ảnh grayscale một kênh (channels: 1) theo định dạng nhãn YOLO:

coco8-grayscale/
├── images/
│   ├── train/   # 4 images
│   └── val/     # 4 images
└── labels/
    ├── train/
    └── val/

Khám phá COCO8-Grayscale trên Ultralytics Platform để xem từng hình ảnh cùng các lớp phủ annotation, xem phân bố class và heatmap bounding box trong tab Charts, rồi clone bộ dữ liệu để huấn luyện model của riêng bạn trên cloud.

YAML của bộ dữ liệu#

Cấu hình bộ dữ liệu COCO8-Grayscale được định nghĩa trong một file YAML, trong đó chỉ định các đường dẫn đến dataset, tên class và metadata thiết yếu khác. Bạn có thể xem file chính thức coco8-grayscale.yaml trong repository Ultralytics trên GitHub.

Lưu ý

Để huấn luyện ảnh RGB của bạn ở chế độ grayscale, bạn chỉ cần thêm channels: 1 vào file YAML của dataset. Thao tác này sẽ chuyển đổi toàn bộ ảnh sang grayscale trong quá trình huấn luyện, cho phép bạn tận dụng các lợi ích của grayscale mà không cần một dataset riêng.

ultralytics/cfg/datasets/coco8-grayscale.yaml
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license

# COCO8-Grayscale dataset (first 8 images from COCO train2017) by Ultralytics
# Documentation: https://docs.ultralytics.com/datasets/detect/coco8-grayscale
# Example usage: yolo train data=coco8-grayscale.yaml
# parent
# ├── ultralytics
# └── datasets
#     └── coco8-grayscale ← downloads here (1 MB)

# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: coco8-grayscale # dataset root dir
train: images/train # train images (relative to 'path') 4 images
val: images/val # val images (relative to 'path') 4 images
test: # test images (optional)

channels: 1

# Classes
names:
  0: person
  1: bicycle
  2: car
  3: motorcycle
  4: airplane
  5: bus
  6: train
  7: truck
  8: boat
  9: traffic light
  10: fire hydrant
  11: stop sign
  12: parking meter
  13: bench
  14: bird
  15: cat
  16: dog
  17: horse
  18: sheep
  19: cow
  20: elephant
  21: bear
  22: zebra
  23: giraffe
  24: backpack
  25: umbrella
  26: handbag
  27: tie
  28: suitcase
  29: frisbee
  30: skis
  31: snowboard
  32: sports ball
  33: kite
  34: baseball bat
  35: baseball glove
  36: skateboard
  37: surfboard
  38: tennis racket
  39: bottle
  40: wine glass
  41: cup
  42: fork
  43: knife
  44: spoon
  45: bowl
  46: banana
  47: apple
  48: sandwich
  49: orange
  50: broccoli
  51: carrot
  52: hot dog
  53: pizza
  54: donut
  55: cake
  56: chair
  57: couch
  58: potted plant
  59: bed
  60: dining table
  61: toilet
  62: tv
  63: laptop
  64: mouse
  65: remote
  66: keyboard
  67: cell phone
  68: microwave
  69: oven
  70: toaster
  71: sink
  72: refrigerator
  73: book
  74: clock
  75: vase
  76: scissors
  77: teddy bear
  78: hair drier
  79: toothbrush

# Download script/URL (optional)
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/coco8-grayscale.zip

Cách sử dụng#

Bộ dữ liệu COCO8-Grayscale (1 MB) sẽ tự động được tải xuống khi bạn bắt đầu huấn luyện lần đầu. Để huấn luyện model YOLO26n trên COCO8-Grayscale trong 100 epoch với kích thước ảnh 640, hãy sử dụng các ví dụ sau. Để xem danh sách đầy đủ các tùy chọn huấn luyện, hãy tham khảo tài liệu Huấn luyện YOLO.

Ví dụ huấn luyện
from ultralytics import YOLO

# Load a pretrained YOLO26n model
model = YOLO("yolo26n.pt")

# Train the model on COCO8-Grayscale
results = model.train(data="coco8-grayscale.yaml", epochs=100, imgsz=640)

Hình ảnh mẫu và annotation#

Dưới đây là một ví dụ về batch huấn luyện ghép mosaic từ bộ dữ liệu COCO8-Grayscale:

COCO8 grayscale dataset mosaic training batch
  • Ảnh ghép mosaic: Ảnh này minh họa một batch huấn luyện trong đó nhiều ảnh từ bộ dữ liệu được kết hợp bằng phép tăng cường mosaic. Phép tăng cường mosaic làm tăng tính đa dạng của các đối tượng và cảnh trong mỗi batch, giúp model khái quát tốt hơn với nhiều kích thước đối tượng, tỷ lệ khung hình và nền khác nhau.

Kỹ thuật này đặc biệt hữu ích cho các dataset nhỏ như COCO8-Grayscale vì tận dụng tối đa giá trị của từng hình ảnh trong quá trình huấn luyện.

Trích dẫn và ghi nhận đóng góp#

Nếu bạn sử dụng bộ dữ liệu COCO trong nghiên cứu hoặc phát triển, vui lòng trích dẫn bài báo sau:

Trích dẫn
@misc{lin2015microsoft,
      title={Microsoft COCO: Common Objects in Context},
      author={Tsung-Yi Lin and Michael Maire and Serge Belongie and Lubomir Bourdev and Ross Girshick and James Hays and Pietro Perona and Deva Ramanan and C. Lawrence Zitnick and Piotr Dollár},
      year={2015},
      eprint={1405.0312},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

Đặc biệt cảm ơn COCO Consortium vì những đóng góp liên tục cho cộng đồng thị giác máy tính.

FAQ#

  • Bộ dữ liệu Ultralytics COCO8-Grayscale được thiết kế để kiểm thử nhanh và debug các model phát hiện đối tượng. Với chỉ 8 hình ảnh (4 ảnh huấn luyện và 4 ảnh validation), bộ dữ liệu này lý tưởng để kiểm tra pipeline huấn luyện YOLO và bảo đảm mọi thứ hoạt động như mong đợi trước khi mở rộng lên các dataset lớn hơn. Hãy khám phá cấu hình YAML của COCO8-Grayscale để biết thêm chi tiết.

  • Bạn có thể huấn luyện model YOLO26 trên COCO8-Grayscale bằng Python hoặc CLI:

    Ví dụ huấn luyện
    from ultralytics import YOLO
    
    # Load a pretrained YOLO26n model
    model = YOLO("yolo26n.pt")
    
    # Train the model on COCO8-Grayscale
    results = model.train(data="coco8-grayscale.yaml", epochs=100, imgsz=640)

    Để biết thêm các tùy chọn huấn luyện, hãy tham khảo tài liệu Huấn luyện YOLO.

  • Augmentation mosaic, được sử dụng trong quá trình huấn luyện COCO8-Grayscale, kết hợp nhiều hình ảnh thành một ảnh trong mỗi batch. Điều này làm tăng tính đa dạng của các đối tượng và background, giúp model YOLO của bạn tổng quát hóa tốt hơn với các tình huống mới. Augmentation mosaic đặc biệt có giá trị đối với các dataset nhỏ vì tận dụng tối đa thông tin có sẵn trong mỗi bước huấn luyện. Để tìm hiểu thêm, hãy xem hướng dẫn huấn luyện.

  • Để validation model YOLO26 sau khi huấn luyện trên COCO8-Grayscale, hãy sử dụng các lệnh validation của model trong Python hoặc CLI. Thao tác này đánh giá hiệu suất của model bằng các metric tiêu chuẩn. Để xem hướng dẫn từng bước, hãy truy cập tài liệu Validation YOLO.

Bình luận