MMFashion — это опенсорсный инструмент для анализа изображений одежды, который основан на PyTorch. Библиотека разрабатывается исследователями из Multimedia Lab в CUHK.
Подробнее про библиотеку
Разработчики выделяют следующие характеристики библиотеки:
- Гибкость, которая обеспечивается засчет модульной архитектуры библиотеки и простоты расширения;
- Доступность готовых моделей для пользователей без специализированных знаний в глубоком обучении;
- Поддержка широкого спектра задач в сфере анализа одежды
В список задач, которые поддерживает MMFashion входят:
- Предсказания типа одежды;
- Распознавание и извлечение одежды на изображении;
- Распознавание структуры одежды (Fashion Landmark Detection);
- Парсинг и сегментация одежды;
- Рекомендация одежды;
- Виртуальная примерка одежды
Разработчики предоставляют тестовые скрипты для оценки датасетов. Кроме того, в MMFashion доступны высокоуровневые APIs для интеграции моделей из библиотеки в существующие пайплайны.
Чтобы использовать кастомный датасет, его необходимо привести к одному из стандартизированных форматов: AttrDataset, InShopDataset, ConsumerToShopDataset or LandmarkDetectDataset.
