svlipatov
Update README.md
70ee9cc
|
raw
history blame
983 Bytes

proj

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

  1. check_photo_model_retrain.py - файл для переобучения последнего слоя модели googlenet обученной на датасете IMAGENET1K_V1.
  2. pikle_model.pkl - сохранненная в файл переобученная модель из п.1. Обучена на подготовленном датасете
  3. check_photo.py - запуск распознования фотографии моделью. Возращает определнную категорию и вероятность в долях.
  4. test_check_photo.py - Запуск распознования фото на файлах из тестовой выборки (в папке Test_photo)
  5. check_photo_model_init.py инициализация модели. Модель поднимается из файла pickle_model.pkl, категории полнимаются из файла cat.csv