search

TianzhongSong / OCR_REST_api

深度学习OCR REST api (Flask+Redis+Keras)
12 stars 8 forks source link

模型加载慢问题 #1

Open sportzhang opened 4 years ago

sportzhang commented 4 years ago

你好，请问一下server中模型的加载不会比较慢吗？

def ocr_process():
    # 载入ocr模型
    model = ocr.Ocr(text_process=False)
    print("load model done")

每次调用函数都要model = ocr.Ocr(text_process=False)，这样会降低效率吧？有没有什么优化方式呢？因为我也遇到了同样的问题。想将model = ocr.Ocr(text_process=False)放到函数外部加载为全局变量，但发现有问题，不知道有什么解决方式？

guanshuicheng commented 4 years ago

定义个推理class，把model load放到init，再定义一个run执行推理