Предварительная загрузка модели yolo в AppConfig Django (Попытка использовать закрытую сессию)

У меня есть модель для обнаружения объектов с помощью YOLO3, и я пытаюсь оптимизировать ее, требуется 10 секунд для загрузки модели, а затем он начинает обнаружение объекта, поэтому я попытался предварительно загрузить модель в apps.py, и передать ее в функцию, которой нужна модель, но я получаю Attemped to use a closed Session from 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

Возможно, есть другой способ сделать это более чистым способом, так что скажите мне, если я ошибаюсь .

Вернуться на верх