AIIJC. 1 место в треке "ИИ в медицине"

Данные

• MosMedData: результаты исследований компьютерной томографии органов грудной клетки с признаками COVID-19
• MedSeg: COVID-19 CT segmentation dataset
• Zenodo: COVID-19 CT Lung and Infection Segmentation Dataset

Запуск кода

Для запуска сайта: streamlit run app.py

Для инференса: python3 inference.py -h

Структура проекта

• app.py - код сайта
• main.py - главный модуль, запускается для тренировки
• config.py - файл конфигураций
• data_functions.py - модули подготовки данных для обучения
• train_functions.py - модули тренировки и валидации
• productions.py - методы подготовки тестовых данных для предсказания
• utils.py - вспомогательные модули
• custom - содержит кастомные модули
  • losses.py - кастомные функции потерь
  • metrics.py - кастомные метрики
  • models.py - кастомные классы моделей
• requirements.txt - файл зависимостей
• checkpoints
  • Binary.pth
  • MultiClass.pth
  • Lungs.pth

Папку checkpoints скачать тут

Тренировка происходила на Google Colab Pro