YOLO Vision 2026:
No license

Набор данных для сегментации упаковок#

Open Package Segmentation Dataset In Colab

Датасет сегментации упаковок Ultralytics представляет собой тщательно подобранную коллекцию из 2197 размеченных изображений упаковок для обучения моделей сегментации экземпляров на одном классе package. Созданный для логистики и автоматизации складов, таких как идентификация, сортировка и обработка упаковок, он напрямую сочетается с Ultralytics YOLO для анализа упаковок в реальном времени в конвейерах компьютерного зрения. Ознакомься с другими датасетами сегментации на нашей странице обзора датасетов.



Watch: Train a Package Segmentation Model using Ultralytics YOLO | Industrial Packages 🎉

Структура набора данных#

Набор данных Package Segmentation распределяет свои 2197 изображений следующим образом:

  • Обучающий набор: 1920 изображений, используемых для обучения модели глубокого обучения.
  • Набор валидации: 188 изображений, используемых во время обучения для настройки гиперпараметров и предотвращения переобучения.
  • Testing set: 89 изображений, отложенных для оценки модели после обучения.
  • Классы: один класс package, охватывающий каждую размеченную упаковку.
  • Download size: ~103 МБ.

Применение#

Сегментация упаковок оптимизирует логистику, доставку «последней мили», контроль качества на производстве и системы «умного города», с применением в электронной коммерции и проверках безопасности. Точные маски упаковок позволяют автоматизированным системам находить, считать и осматривать посылки в режиме реального времени.

Умные склады и логистика#

На современных склах решения на базе визуального ИИ могут оптимизировать операции за счет автоматизации идентификации и сортировки упаковок. Модели компьютерного зрения, обученные на этом датасете, могут быстро обнаруживать и сегментировать упаковки в реальном времени даже в сложных условиях при слабом освещении или в загроможденных пространствах. Это приводит к сокращению времени обработки, уменьшению количества ошибок и повышению общей эффективности логистических операций.

Контроль качества и обнаружение повреждений#

Модели сегментации упаковок могут определять поврежденные упаковки, анализируя их форму и внешний вид. Обнаруживая неровности или деформации в контурах упаковок, эти модели помогают гарантировать, что через цепочку поставок проходят только целые упаковки, уменьшая количество жалоб клиентов и уровень возвратов. Это ключевой аспект контроля качества в производстве и жизненно важен для поддержания целостности продукта.

Полный датасет сегментации упаковок также можно просматривать и управлять им на платформе Ultralytics.

YAML набора данных#

Файл YAML определяет конфигурацию датасета, включая пути, классы и другие важные детали. Для датасета сегментации упаковок файл package-seg.yaml поддерживается по адресу https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/package-seg.yaml.

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

# Package-seg dataset by Ultralytics
# Documentation: https://docs.ultralytics.com/datasets/segment/package-seg
# Example usage: yolo train data=package-seg.yaml
# parent
# ├── ultralytics
# └── datasets
#     └── package-seg ← downloads here (103 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: package-seg # dataset root dir
train: images/train # train images (relative to 'path') 1920 images
val: images/val # val images (relative to 'path') 188 images
test: images/test # test images (relative to 'path') 89 images

# Classes
names:
  0: package

# Download script/URL (optional)
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/package-seg.zip

Использование#

Чтобы обучить модель Ultralytics YOLO26n на датасете сегментации упаковок в течение 100 эпох с размером изображения 640, используй следующие фрагменты кода. Датасет (~103 МБ) загружается автоматически при первом использовании. Полный список доступных аргументов см. на странице модели Обучение.

Пример обучения
from ultralytics import YOLO

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

# Train the model on the Package Segmentation dataset
results = model.train(data="package-seg.yaml", epochs=100, imgsz=640)

# Validate the model
results = model.val()

# Perform inference on an image
results = model("path/to/image.jpg")

Примеры данных и разметки#

Ниже приведен пример из набора данных Package Segmentation с наложенными масками сегментации, очерчивающими обнаруженные упаковки:

Образец датасета сегментации упаковок для логистики

Датасет охватывает самые разные локации, среды и плотности упаковок, поэтому обученные на нем модели видят все необходимые реальные логистические сцены для генерализации. См. страницу задачи сегментации для ознакомления со связанными рабочими процессами.

Цитирование и благодарности#

Если ты используешь набор данных Package Segmentation в своих исследовательских или девелоперских инициативах, пожалуйста, правильно укажи источник:

Цитата
@misc{ factory_package_dataset,
    title = { factory_package Dataset },
    type = { Open Source Dataset },
    author = { factorypackage },
    url = { https://universe.roboflow.com/factorypackage/factory_package },
    year = { 2024 },
    month = { jan },
    note = { visited on 2024-01-24 },
}

Мы выражаем благодарность создателям датасета сегментации упаковок за их вклад в сообщество компьютерного зрения. Чтобы найти больше датасетов, посети коллекцию датасетов Ultralytics и наше руководство по советам по обучению моделей.

FAQ#

  • Датасет сегментации упаковок представляет собой коллекцию из 2197 размеченных изображений упаковок для обучения и оценки моделей сегментации экземпляров на одном классе package. Он нацелен на логистику и приложения автоматизации складов, такие как идентификация упаковок, сортировка и контроль качества, и используется напрямую с Ultralytics YOLO26 через файл конфигурации package-seg.yaml.

  • Всего в датасете 2197 изображений: 1920 для обучения, 188 для валидации и 89 для тестирования — все они размечены для одного класса package. Полный архив загружается автоматически в виде архива .zip размером около ~103 МБ при первом использовании.

  • Загрузи предобученную модель сегментации (например, yolo26n-seg.pt) и обучи ее с конфигурацией package-seg.yaml, используя фрагменты кода для Python или CLI в разделе Использование выше. Полный список доступных аргументов см. в руководстве по обучению.

  • YOLO26 обеспечивает передовую точность и скорость реального времени для сегментации экземпляров, позволяя автоматизированным системам надежно обнаруживать и сортировать упаковки даже на тусклых или загроможденных склах — см. раздел Приложения выше. Обученные модели экспортируются в такие форматы, как ONNX и TensorRT, для развертывания на складском оборудовании.

  • Файл package-seg.yaml, который определяет пути к датасету и единственный класс package, находится в репозитории Ultralytics на GitHub: package-seg.yaml.

Комментарии