File size: 1,333 Bytes
adfb4d5
 
db33c51
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
# proj
Smart city gid
В приложенных файлах:
1. Model_retrain.py - файл для переобучения последнего слоя модели googlenet обученной на датасете IMAGENET1K_V1. Датасет для переобучения скачивал по этой ссылке 'https://drive.google.com/uc?export=download&id=120xqh0mYtYZ1Qh7vr-XFzjPbSKivLJjA'
2. pikle_model.pkl  - сохранненная в файл переобученная модель из п.1
3. model_execute - запуск модели сохранненной в файл на нескольких фотграфиях
4. Скриншот результата выполнения
5. Фото для теста

Проблема: Не совсем понятно как интерпретировать результат. Если в результате выводятся вероятности соответствия фотографии каждой из категорий, то непонятно как идентифицировать категории из результата. Так же это будет значить что модель на текущий момент работает плохо. Возможно что-то не так в процессе переобучения.