Spaces:

molokhovdmitry
/

landmark_recognition

Sleeping

File size: 1,177 Bytes

adfb4d5
 
db33c51
70ee9cc
 
 
 
 
62390f4

# 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
6. wikipedia_api.py - поиск на Википедия. Принимается список для поиска, возвращается список с результатами поиска