Spaces:
Sleeping
Sleeping
# 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. Фото для теста | |
Проблема: Не совсем понятно как интерпретировать результат. Если в результате выводятся вероятности соответствия фотографии каждой из категорий, то непонятно как идентифицировать категории из результата. Так же это будет значить что модель на текущий момент работает плохо. Возможно что-то не так в процессе переобучения. | |