#python #django #tensorflow #session #yolo
Вопрос:
Итак , у меня есть модель для обнаружения объектов с использованием YOLO3 , и я пытаюсь ее оптимизировать, загрузка модели занимает 10 секунд, а затем он запускает обнаружение объектов, поэтому я попытался предварительно загрузить модель в apps.py , и передайте его функции , которой нужна модель , но я пытаюсь использовать закрытый сеанс из tensorflow, есть ли способ закрыть сеанс в другом месте, откуда загружается модель ? на самом деле это то, что я делаю . apps.py
from django.apps import AppConfig import snakeimage.classification_codes.classification_codes.pre_loading as pre import tensorflow init_tf = None class SnakeimageConfig(AppConfig): default_auto_field = 'django.db.models.BigAutoField' name = 'snakeimage' def ready(self): with tensorflow.compat.v1.Session(): print("im ready !!") global init_tf init_tf = pre.loading_model() и я использую глобальную переменную в этой функции :
import snakeimage.apps as si import tensorflow def detection_task(image_path): obj_det_res = snake_obj_detection(image_path , si.init_tf) print(obj_det_res) return obj_det_res может быть , у них есть другой способ сделать это более чистым способом, поэтому скажите мне, если я ошибаюсь .