Лабораторная работа 0.1 (Знакомство с библиотеками timm и smp)
Требуемые условия завершения
Дополнительно: минимальный пайлпайн для обучения модели из smp
Открыто с: понедельник, 16 сентября 2024, 00:00
Срок сдачи: среда, 2 октября 2024, 00:00
Задания:
- Установите библиотеку timm, следуя следующей инструкции: https://github.com/huggingface/pytorch-image-models
- Познакомтесь с "пулом" моделей timm: https://timm.fast.ai/
- Создайте любую модель из timm.
- Напишите код запуска модели на пользовательском изображении
- Найдите в интернете список с названиями классов, на которых обучалась модель и реализуйте преобразование индекса предсказанного класса в его название
- Оттрасируйте с помощью torch.jit.trace модель и сохраните её
- Установите библиотеку SMP
- Ознакомьтесь с документацией проекта: https://smp.readthedocs.io/en/latest/
- Создайте модель с помощью SMP с encoder частью из timm (список поддерживаемых timm моделей)
- Попробуйте создать, загрузив веса предобученной модели на двух разных наборах данных (требуется создать две разных модели для этого), список доступных весов для разных наборов данных
- Сохраните веса (state_dict) encoder части сегментационной модели в отдельный файл
- Загрузите веса сохраненной модели через экземпляр класса модели, созданной с помощью timm