Набор данных для сегментации упаковок#
Датасет сегментации упаковок 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 🚀 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.



