Meet YOLO26: next-gen vision AI.

Link to this sectionBộ dữ liệu COCO128-Seg#

Link to this sectionGiới thiệu#

Ultralytics COCO128-Seg là một bộ dữ liệu instance segmentation nhỏ gọn nhưng linh hoạt, bao gồm 128 hình ảnh đầu tiên của tập COCO train 2017. Bộ dữ liệu này rất lý tưởng để kiểm thử và gỡ lỗi các model segmentation, hoặc để thử nghiệm các hướng tiếp cận phát hiện mới. Với 128 hình ảnh, nó đủ nhỏ để dễ dàng quản lý, nhưng vẫn đủ đa dạng để kiểm tra các pipeline huấn luyện tìm lỗi và đóng vai trò kiểm tra tính hợp lệ (sanity check) trước khi huấn luyện trên các bộ dữ liệu lớn hơn.

Link to this sectionCấu trúc tập dữ liệu#

  • Hình ảnh: Tổng cộng 128. Tệp YAML mặc định sử dụng lại cùng một thư mục cho cả train và val để bạn có thể lặp lại nhanh chóng, nhưng bạn có thể sao chép hoặc tùy chỉnh phân đoạn nếu muốn.
  • Lớp (Classes): Gồm 80 danh mục đối tượng tương tự như COCO.
  • Nhãn (Labels): Các đa giác định dạng YOLO được lưu bên cạnh mỗi hình ảnh trong labels/{train,val}.

Tập dữ liệu này được thiết kế để sử dụng với Ultralytics PlatformYOLO26.

Link to this sectionYAML tập dữ liệu#

Một tệp YAML (Yet Another Markup Language) được sử dụng để xác định cấu hình bộ dữ liệu. Nó chứa thông tin về đường dẫn, các lớp của bộ dữ liệu và các thông tin liên quan khác. Trong trường hợp của bộ dữ liệu COCO128-Seg, tệp coco128-seg.yaml được duy trì tại https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/coco128-seg.yaml.

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

# COCO128-seg dataset https://www.kaggle.com/datasets/ultralytics/coco128 (first 128 images from COCO train2017) by Ultralytics
# Documentation: https://docs.ultralytics.com/datasets/segment/coco128-seg
# Example usage: yolo train data=coco128-seg.yaml
# parent
# ├── ultralytics
# └── datasets
#     └── coco128-seg ← downloads here (7 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: coco128-seg # dataset root dir
train: images/train2017 # train images (relative to 'path') 128 images
val: images/train2017 # val images (relative to 'path') 128 images
test: # test images (optional)

# 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/coco128-seg.zip

Link to this sectionCách sử dụng#

Để huấn luyện một model YOLO26n-seg trên bộ dữ liệu COCO128-Seg trong 100 epoch với kích thước ảnh là 640, bạn có thể sử dụng các đoạn code sau. Để có danh sách đầy đủ các đối số khả dụng, hãy tham khảo trang Training của model.

Ví dụ về Training
from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n-seg.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="coco128-seg.yaml", epochs=100, imgsz=640)

Link to this sectionHình ảnh mẫu và chú thích#

Dưới đây là một số ví dụ về các hình ảnh từ bộ dữ liệu COCO128-Seg, cùng với các chú thích tương ứng của chúng:

COCO128-seg instance segmentation dataset mosaic
  • Hình ảnh Mosaiced: Hình ảnh này minh họa một batch huấn luyện bao gồm các hình ảnh từ tập dữ liệu đã được áp dụng kỹ thuật Mosaic. Mosaic là một kỹ thuật được sử dụng trong quá trình huấn luyện nhằm kết hợp nhiều hình ảnh thành một hình ảnh duy nhất để tăng sự đa dạng của các đối tượng và bối cảnh trong mỗi batch huấn luyện. Điều này giúp cải thiện khả năng tổng quát hóa của model đối với các kích thước đối tượng, tỷ lệ khung hình và bối cảnh khác nhau.

Ví dụ minh họa sự đa dạng và phức tạp của các hình ảnh trong bộ dữ liệu COCO128-Seg và những lợi ích của việc sử dụng kỹ thuật mosaicing trong quá trình huấn luyện.

Link to this sectionTrích dẫn và Ghi nhận#

Nếu bạn sử dụng bộ dữ liệu COCO trong công việc nghiên cứu hoặc phát triển của mình, 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}
}

Chúng tôi xin ghi nhận COCO Consortium vì đã tạo ra và duy trì nguồn tài nguyên quý giá này cho cộng đồng computer vision. Để biết thêm thông tin về tập dữ liệu COCO và những người tạo ra nó, hãy truy cập trang web tập dữ liệu COCO.

Link to this sectionCâu hỏi thường gặp#

Link to this sectionBộ dữ liệu COCO128-Seg là gì và nó được sử dụng như thế nào trong Ultralytics YOLO26?#

Bộ COCO128-Seg dataset là một bộ dữ liệu instance segmentation nhỏ gọn do Ultralytics cung cấp, bao gồm 128 hình ảnh đầu tiên từ tập COCO train 2017. Bộ dữ liệu này được thiết kế để kiểm thử và gỡ lỗi các model segmentation hoặc thử nghiệm các phương pháp phát hiện mới. Nó đặc biệt hữu ích với Ultralytics YOLO26Platform để lặp lại nhanh chóng và kiểm tra lỗi pipeline trước khi mở rộng quy mô sang các bộ dữ liệu lớn hơn. Để biết cách sử dụng chi tiết, hãy tham khảo trang Training của model.

Link to this sectionLàm cách nào để tôi có thể huấn luyện một model YOLO26n-seg bằng cách sử dụng bộ dữ liệu COCO128-Seg?#

Để huấn luyện một model YOLO26n-seg trên bộ dữ liệu COCO128-Seg trong 100 epoch với kích thước ảnh là 640, bạn có thể sử dụng Python hoặc các lệnh CLI. Đây là một ví dụ nhanh:

Ví dụ về Training
from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n-seg.pt")  # Load a pretrained model (recommended for training)

# Train the model
results = model.train(data="coco128-seg.yaml", epochs=100, imgsz=640)

Để có lời giải thích thấu đáo về các đối số khả dụng và các tùy chọn cấu hình, bạn có thể xem tài liệu Training.

Link to this sectionTại sao bộ dữ liệu COCO128-Seg lại quan trọng đối với việc phát triển và gỡ lỗi model?#

Bộ COCO128-Seg dataset mang lại sự kết hợp cân bằng giữa khả năng quản lý và sự đa dạng với 128 hình ảnh, giúp nó trở nên hoàn hảo để kiểm thử và gỡ lỗi nhanh chóng các model segmentation hoặc thử nghiệm các kỹ thuật phát hiện mới. Kích thước vừa phải của nó cho phép thực hiện các vòng lặp huấn luyện nhanh chóng, đồng thời cung cấp đủ sự đa dạng để xác thực các pipeline huấn luyện trước khi mở rộng sang các bộ dữ liệu lớn hơn. Tìm hiểu thêm về các định dạng bộ dữ liệu được hỗ trợ trong hướng dẫn về bộ dữ liệu segmentation của Ultralytics.

Link to this sectionTôi có thể tìm thấy tệp cấu hình YAML cho bộ dữ liệu COCO128-Seg ở đâu?#

Tệp cấu hình YAML cho COCO128-Seg dataset có sẵn trong kho lưu trữ của Ultralytics. Bạn có thể truy cập trực tiếp vào tệp tại https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/coco128-seg.yaml. Tệp YAML bao gồm thông tin cần thiết về đường dẫn bộ dữ liệu, các lớp và các thiết lập cấu hình cần thiết để huấn luyện và xác thực model.

Link to this sectionMột số lợi ích của việc sử dụng mosaicing trong quá trình huấn luyện với bộ dữ liệu COCO128-Seg là gì?#

Sử dụng mosaicing trong quá trình huấn luyện giúp tăng tính đa dạng và phong phú của các đối tượng và cảnh trong mỗi batch huấn luyện. Kỹ thuật này kết hợp nhiều hình ảnh thành một hình ảnh tổng hợp duy nhất, tăng cường khả năng tổng quát hóa của model đối với các kích thước đối tượng, tỷ lệ khung hình và bối cảnh khác nhau trong cảnh. Mosaicing có lợi cho việc cải thiện độ bền vững và độ chính xác của model, đặc biệt khi làm việc với các bộ dữ liệu có kích thước vừa phải như COCO128-Seg. Để xem ví dụ về các hình ảnh được mosaiced, hãy xem phần Sample Images and Annotations.

Người đóng góp

Bình luận