Flask服务器启动后自动加载

时间:2019-12-03 01:57:15

标签: tensorflow flask

我正在建立一个用于图像检测的网站,我想在启动服务器后立即加载并启动pb数据。这是因为缓存使检测速度更快。

有什么好方法吗?

output_detect = tf_module.detect(image)

当前tf_module在图像上传后运行。 我可以在不上传图片的情况下运行模块吗?

1 个答案:

答案 0 :(得分:0)

一个简单的解决方案是在第一个请求装饰器之前实现。用@ app.before_first_request装饰的函数将在对该应用程序实例的第一个请求之前运行一次。启动服务器后,您可以自己触发此第一个请求。小例子:

@app.before_first_request
def before_first_request_func():
    print("This function runs once")