Ultralytics YOLO27:
Get Started

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 được chuyển sang định dạng grayscale — 4 hình ảnh dùng để huấn luyện và 4 hình ảnh để xác thực. Bộ dữ liệu này được thiết kế dành riêng cho việc kiểm thử nhanh, gỡ lỗi và thử nghiệm với các model grayscale YOLO cùng pipeline huấn luyện. Kích thước nhỏ giúp bộ dữ liệu dễ quản lý, trong khi tính đa dạng khiến bộ dữ liệu trở thành công cụ kiểm tra tính hợp lý hiệu quả trước khi mở rộng quy mô sang các bộ dữ liệu lớn hơn.



Xem: Cách huấn luyện Ultralytics YOLO trên dataset ảnh xám và đa phổ | VisionAI đa kênh 🚀

Cấu trúc dataset#

COCO8-Grayscale có cấu trúc tương tự COCO8 — gồm 8 hình ảnh đầu tiên từ tập COCO train 2017, chia thành 4 hình ảnh dùng để huấn luyện và 4 hình ảnh để xác thực, được gán nhãn bằng 80 tên lớp COCO — nhưng mỗi hình ả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 để duyệt từng hình ảnh cùng lớp phủ chú thích, xem phân bố lớp và heatmap bounding box trong tab Charts, rồi sao chép bộ dữ liệu để huấn luyện model của riêng bạn trên cloud.

YAML của dataset#

Cấu hình bộ dữ liệu COCO8-Grayscale được xác định trong tệp YAML, trong đó chỉ định đường dẫn bộ dữ liệu, tên lớp và metadata thiết yếu khác. Bạn có thể xem tệp coco8-grayscale.yaml chính thức trong kho GitHub của Ultralytics.

Lưu ý

Để huấn luyện ảnh RGB ở dạng grayscale, bạn chỉ cần thêm channels: 1 vào tệp YAML của bộ dữ liệu. Thao tác này chuyển tất cả hình ảnh sang grayscale trong quá trình huấn luyện, cho phép bạn tận dụng lợi ích của grayscale mà không cần bộ dữ liệu 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 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

# Tải model YOLO26n đã được huấn luyện trước
model = YOLO("yolo26n.pt")

# Huấn luyện model trên COCO8-Grayscale
results = model.train(data="coco8-grayscale.yaml", epochs=100, imgsz=640)

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

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

COCO8 grayscale dataset mosaic training batch
  • Ảnh ghép Mosaic: Hình ảnh này minh họa một batch huấn luyện trong đó nhiều hình ảnh của bộ dữ liệu được kết hợp bằng mosaic augmentation. Mosaic augmentation làm tăng sự đa dạng của đối tượng và bối cảnh trong mỗi batch, giúp model tổng quát hóa 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 với các bộ dữ liệu nhỏ như COCO8-Grayscale, vì tối đa hóa giá trị của mỗi hình ảnh trong quá trình huấn luyện.

Trích dẫn và lời cảm ơn#

Nếu 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}
}

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

Câu hỏi thường gặp#

  • Bộ dữ liệu COCO8-Grayscale của Ultralytics được thiết kế để kiểm thử nhanh và gỡ lỗi các model phát hiện đối tượng. Với chỉ 8 hình ảnh (4 hình ảnh dùng để huấn luyện, 4 hình ảnh để xác thực), bộ dữ liệu này rất phù hợp để kiểm tra pipeline huấn luyện YOLO và đảm bảo mọi thứ hoạt động như mong đợi trước khi mở rộng quy mô sang các bộ dữ liệu lớn hơn. 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
    
    # Tải model YOLO26n đã được huấn luyện trước
    model = YOLO("yolo26n.pt")
    
    # Huấn luyện model trên 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.

  • Mosaic augmentation đượ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 trong mỗi batch. Kỹ thuật này làm tăng sự đa dạng của đối tượng và nền, giúp model YOLO tổng quát hóa tốt hơn với các tình huống mới. Mosaic augmentation đặc biệt hữu ích với các bộ dữ liệu nhỏ vì tối đa hóa lượng 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.

  • Để xác thực model YOLO26 sau khi huấn luyện trên COCO8-Grayscale, hãy dùng các lệnh xác thực của model trong Python hoặc CLI. Thao tác này đánh giá hiệu năng 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 Xác thực YOLO.

Bình luận