svlipatov
Update README.md
db33c51
|
raw
history blame
1.33 kB
# 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. Фото для теста
Проблема: Не совсем понятно как интерпретировать результат. Если в результате выводятся вероятности соответствия фотографии каждой из категорий, то непонятно как идентифицировать категории из результата. Так же это будет значить что модель на текущий момент работает плохо. Возможно что-то не так в процессе переобучения.