flask服务器启动后立即自动加载

Question

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

有什么好办法吗？

output_detect = tf_module.detect(image)

目前tf_module是在镜像上传后运行的。我可以不上传镜像就运行模块吗？

Answer 1

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

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