YOLO Vision 2026:

Dataset COCO128-Seg#

Introdução#

Ultralytics COCO128-Seg é um conjunto de dados de segmentação de instâncias pequeno, mas versátil, composto pelas primeiras 128 imagens do conjunto COCO train 2017. Este conjunto de dados é ideal para testar e depurar modelos de segmentação, ou para experimentar novas abordagens de deteção. Com 128 imagens, é suficientemente pequeno para ser facilmente gerível, mas diversificado o suficiente para testar pipelines de treino em busca de erros e servir como uma verificação de sanidade antes de treinar conjuntos de dados maiores.

Estrutura do Dataset#

  • Imagens: 128 no total, com divisão de train e val idêntica (veja a nota abaixo).
  • Classes: As mesmas 80 categorias de objetos do COCO.
  • Rótulos: Polígonos no formato YOLO armazenados em labels/train2017 para o diretório partilhado de imagens de treino e validação.
  • Tamanho do download: ~7 MB.
Nota

O YAML padrão aponta train e val para as mesmas 128 imagens, portanto, as métricas de validação medem o ajuste no conjunto de treinamento em vez da generalização em dados mantidos fora do treino. Duplique ou personalize a divisão se precisar de um conjunto de teste real.

Este conjunto de dados destina-se ao uso com Ultralytics Platform e YOLO26.

YAML do Dataset#

Um ficheiro YAML é utilizado para definir a configuração do conjunto de dados. Ele contém informações sobre os caminhos, as classes e outras informações relevantes do conjunto de dados. No caso do conjunto de dados COCO128-Seg, o ficheiro coco128-seg.yaml é mantido em 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

Uso#

Para treinar um modelo YOLO26n-seg no conjunto de dados COCO128-Seg durante 100 épocas com um tamanho de imagem de 640, podes utilizar os seguintes snippets de código. Para uma lista abrangente de argumentos disponíveis, consulta a página de Treino do modelo.

Exemplo de Treinamento
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)

Exemplos de Imagens e Anotações#

Aqui estão alguns exemplos de imagens do dataset COCO128-Seg, juntamente com as suas respectivas anotações:

COCO128-seg instance segmentation dataset mosaic
  • Imagem em Mosaico: Esta imagem demonstra um lote de treino composto por imagens do conjunto de dados em mosaico. A mosaicação é uma técnica usada durante o treino que combina múltiplas imagens numa única imagem para aumentar a variedade de objetos e cenas dentro de cada lote de treino. Isto ajuda a melhorar a capacidade do modelo de generalizar para diferentes tamanhos de objeto, proporções e contextos.

Citações e Agradecimentos#

Se você utilizar o dataset COCO em seu trabalho de pesquisa ou desenvolvimento, por favor, cite o seguinte artigo:

Citação
@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}
}

Gostaríamos de agradecer ao Consórcio COCO por criar e manter este recurso valioso para a comunidade de computer vision. Para obter mais informações sobre o conjunto de dados COCO e seus criadores, visita o COCO dataset website.

FAQ#

  • O conjunto de dados COCO128-Seg é um conjunto de dados compacto de segmentação de instâncias da Ultralytics, consistindo nas primeiras 128 imagens do conjunto COCO train 2017. Este conjunto de dados foi concebido para testar e depurar modelos de segmentação ou experimentar novos métodos de deteção. É particularmente útil com o YOLO26 e a Platform da Ultralytics para uma iteração rápida e verificação de erros no pipeline antes de passar para conjuntos de dados maiores. Para uma utilização detalhada, consulta a página de Treino do modelo.

  • Para treinar um modelo YOLO26n-seg no dataset COCO128-Seg por 100 épocas com um tamanho de imagem de 640, podes usar comandos Python ou CLI. Aqui está um exemplo rápido:

    Exemplo de Treinamento
    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)

    Para uma explicação detalhada dos argumentos disponíveis e opções de configuração, podes consultar a documentação de Training.

  • Como o download e o ciclo de treino/validação são muito menores do que o COCO completo, o COCO128-Seg permite-te executar uma verificação de sanidade de 1 época num novo pipeline — verificando se o modelo treina, valida e guarda os pontos de controlo corretamente — antes de passar para o conjunto de dados COCO-Seg completo. Sabe mais sobre os formatos de conjuntos de dados suportados no guia de conjuntos de dados de segmentação da Ultralytics.

  • O ficheiro de configuração YAML para o conjunto de dados COCO128-Seg está disponível no repositório da Ultralytics. Podes aceder diretamente ao ficheiro em https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/coco128-seg.yaml. O ficheiro YAML inclui informações essenciais sobre caminhos de conjuntos de dados, classes e definições de configuração necessárias para o treino e validação do modelo.

  • O COCO128-Seg (128 imagens) situa-se entre o COCO8-Seg (8 imagens) e o conjunto de dados COCO-Seg completo (118.287 imagens de treino) em termos de tamanho:

    • COCO8-Seg: 8 imagens (4 de train, 4 de val) — ideal para verificações rápidas e depuração.
    • COCO128-Seg: 128 imagens — equilibrado entre tamanho e diversidade, com treino e val a partilharem o mesmo diretório.
    • COCO-Seg completo: 118.287 imagens de treino — abrangente mas intensivo em termos de recursos, exigindo ~27 GB no primeiro download.

    O COCO128-Seg oferece mais diversidade do que o COCO8-Seg, mantendo-se muito mais gerenciável do que o conjunto de dados COCO-Seg completo para experimentação e desenvolvimento inicial do modelo.

Comentários