proj

Smart city gid В приложенных файлах:

Model_retrain.py - файл для переобучения последнего слоя модели googlenet обученной на датасете IMAGENET1K_V1. Датасет для переобучения скачивал по этой ссылке 'https://drive.google.com/uc?export=download&id=120xqh0mYtYZ1Qh7vr-XFzjPbSKivLJjA'
pikle_model.pkl - сохранненная в файл переобученная модель из п.1
model_execute - запуск модели сохранненной в файл на нескольких фотграфиях
Скриншот результата выполнения
Фото для теста

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