svlipatov commited on
Commit
db33c51
·
1 Parent(s): 87895a7

Update README.md

Browse files

Former-commit-id: 1b8622eaaffc5d6dd45fe2d197d70f76b30ec0f8

Files changed (1) hide show
  1. README.md +8 -0
README.md CHANGED
@@ -1,2 +1,10 @@
1
  # proj
2
  Smart city gid
 
 
 
 
 
 
 
 
 
1
  # proj
2
  Smart city gid
3
+ В приложенных файлах:
4
+ 1. Model_retrain.py - файл для переобучения последнего слоя модели googlenet обученной на датасете IMAGENET1K_V1. Датасет для переобучения скачивал по этой ссылке 'https://drive.google.com/uc?export=download&id=120xqh0mYtYZ1Qh7vr-XFzjPbSKivLJjA'
5
+ 2. pikle_model.pkl - сохранненная в файл переобученная модель из п.1
6
+ 3. model_execute - запуск модели сохранненной в файл на нескольких фотграфиях
7
+ 4. Скриншот результата выполнения
8
+ 5. Фото для теста
9
+
10
+ Проблема: Не совсем понятно как интерпретировать результат. Если в результате выводятся вероятности соответствия фотографии каждой из категорий, то непонятно как идентифицировать категории из результата. Так же это будет значить что модель на текущий момент работает плохо. Возможно что-то не так в процессе переобучения.