В данном репозитории вы увидите решеине кейса "Детектор производственных дефектов" от команды Пёс да лис.
Здесь представлены все необходимые файлы, начиная от кода для формирования датасета и обучения модели, заканчивая ноутбуками для обучения и дообучения модели.
Также здесь представлены 3 файла с моделями, которые наша команда натренировала в рамках хакатона:
Nano_final.pt - самая быстродейственная модель
Medium_final.pt - самая точная модель
Small_final.pt - оптимальный вариант по сочетанию быстродействия и точности
В ноутбуках присутвует весь необходимый код, при помощи которого кейсодерждатель сможет:
- сформировать необходимый для обучения модели датасет
- обучить, либо дообучить модель
- внедрить её в производственный процесс
В данном файле вы можете найти пример интеграции модели в телеграм бота на основе кода из файла Model_usage.ipynb
Также на основе кода из данного файла можно интегрировать модель в любую систему
В файлах confusion_matrix находятся матрицы неточностей, на основе которых можно косвенно судить о качестве работы моделей
Но мы предлагаем Вам лично убедиться в качестве работы наших моделей в Телеграм боте
P.S.
Файл Medium_final.pt не загружается на репозиторий из-за ограничений по размеру, поэтому вот ссылка на него https://drive.google.com/file/d/1_kz32RR8Ks0xdxRGXvDQHfVoU4ke-HYq/view?usp=sharing