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.
Để 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 🚀 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.zipCá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.
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:
- Ả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:
@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ệnfrom 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.



