├── Dockerfile
├── .gitignore
├── infer.py - Запустите, чтобы получить предсказания по тестовым данным
├── petfinder
│ ├── config.py - Конфигурация модели
│ ├── dataset.py - Модуль для работы с данными
│ ├── model.py - Код модели
│ └── util.py - Полезные штуки, сейчас там код для аугментаций
├── README.md
├── requirements.txt
└── train.py - Запустите, чтобы обучить модель
Это kaggle-соревнование. Цель - оценить по картике животного, насколько она окажется популярной в соцсетях.
Ссылка на соревнование и данные https://www.kaggle.com/competitions/petfinder-pawpularity-score/data
Я буду использовать pytorch и предобученную сеть, которую соответственно дообучу для решения задачи. В качестве основы буду использовать Swin Transformer.
Соответственно готовый проект - контейнер с приложением, которому можно скинуть ссылку на изображение и оно вернет оценку (т.е. связь будет осуществляться через некое API по http запросам).
В процессе эксплуатации будет возможность добавить данных и дообучить на них модель, протестировать. Также в процессе эксплуатации будет собираться статистика по использованию.
docker build -t petfinder:0.1 .
далее можно запустить с помощью
docker run --rm -it petfinder:0.1
Прежде чем запускать обучение, скачайте данные
dvc pull