metadata

title: Landmark Recognition
emoji: 😻
colorFrom: red
colorTo: yellow
sdk: streamlit
sdk_version: 1.29.0
app_file: main.py
pinned: false

proj

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

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