Это введение в「SKU110K-DenseDet」, модель машинного обучения, которую можно использовать с ailia SDK. Вы можете легко использовать эту модель для создания приложений ИИ с помощью ailia SDK, а также многих других готовых к использованию AIlia MODELS.
Обзор
SKU110K-DenseDet – это модель машинного обучения, которая может обнаруживать товары в магазине. Он может обнаруживать ограничивающие рамки продуктов, представленных на полке супермаркета, но категоризация отсутствует, обнаруживается только наличие продукта.
О SKU110K
SKU110K – это опубликованный в апреле 2019 года набор данных для обнаружения продуктов, содержащий 11 762 изображения, сделанных мобильными телефонами в тысячах супермаркетов по всему миру (США, Европа, Восточная Азия). Ограничительные рамки были аннотированы вручную. Он содержит 90 968 ограничивающих рамок на 8 233 изображениях для обучения и 432 312 ограничивающих рамок на 2 941 изображении для проверки.
GitHub — например, 4000/SKU110K_CVPR19
Набор данных и кодовая база для CVPR2019 «Точное обнаружение в плотно упакованных сценах [ссылка на документ] Типичное изображение на нашем…github.com »
Архитектура
SKU110K-DenseDet прошел обучение с помощью MMDetection. Он достигает 58,0% mAP с использованием Cascade R-CNN. Магистраль использует ResNXt-101.
Поскольку SKU110K содержит много мелких объектов, обычные архитектуры и входные разрешения для обнаружения объектов недостаточно точны. Поэтому входной размер изображений установлен на 2560x2560.
В средах с малым объемом памяти графического процессора во время обучения используется случайное кадрирование изображений таким образом, что это не оказывает негативного влияния на результаты обучения.
SKU110Kсодержит в среднем 150 ограничивающих рамок на изображение. Это значительно больше, чем у MS COCO, и гиперпараметры по умолчанию не оптимальны. Следовательно, максимальное количество положительных образцов как в RPN, так и в R-CNN-пробоотборнике было скорректировано, чтобы снять ограничения.
Применение
SKU110K-DenseDetможно запустить с помощью ailia SDK с помощью следующей команды. Из-за огромного размера магистрали укажите -e 0
для запуска в режиме ЦП в средах с небольшим объемом памяти графического процессора.
$ python3 sku110k-densedet.py -i input.jpg -e 0
Топор Инк. разработал ailia SDK, который обеспечивает кросс-платформенный быстрый вывод на основе графического процессора.
ax Inc. предоставляет широкий спектр услуг от консалтинга и создания моделей до разработки приложений и SDK на основе ИИ. Не стесняйтесь связаться с нами для любого запроса.