landmark_recognition / check_photo_model_init.py
Sergei
Переобученная модель + файл саспознания переделан, для запуска из другого файла
8f186c9
raw
history blame
479 Bytes
import pickle
import csv
# Файл инициализации модели
def init_model():
# Загрузить модели из файла
pkl_filename = "pickle_model.pkl"
with open(pkl_filename, 'rb') as file:
model = pickle.load(file)
# Считывание категорий
file = open("cat.csv", "r")
cat1 = list(csv.reader(file, delimiter=","))
categorias = cat1[0]
file.close()
model.eval()
return model, categorias