lolpa1n / niias_solution

Решение чемпионата от НИИАС

Запуск предикта для тестовых данных

python -m pip install --upgrade pip
pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
pip install -r requirements.txt

Для запуска предикта для тестового датасета необходимо указать путь до папки с тестовыми изображениями в переменную ROOT в ноутбуке predict.ipynb

Загрузка моделей

Модели необходимо загрузить с google-drive в папку models https://drive.google.com/drive/folders/1BYoDgAcN52RTficHY5GQMW5N5MX4L-aJ?usp=sharing

Ссылка на презентацию

https://docs.google.com/presentation/d/1YHGxAtbr3Lrbidk6qJ8jtRC_Kvppf9ZE6tqvuIlxPqU/edit?usp=sharing

Связь

При возникновении проблем при запуске можно связаться по телеграмму - @lolpa1nnn