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



