在 Django 应用程序中加载 TensorFlow 模型的位置

Question

我正在开发一个 Django 项目，该项目应该从文件中加载已保存的 Tensorflow 模型。我应该在哪里运行此代码以获得全局模型变量？目前，我只是在所有函数之外的views.py文件中加载模型并且它正在工作，但我不确定它是否只运行一次以及它是否是一个干净的解决方案。

Answer 1

我建议您开始使用 mlops，并尽可能多地预加载所有内容，这样当您调用 api 时，django 只负责生成预测。我把你的项目留在这里，我有几个 ML/Django/Tflite。

模型可以在媒体文件中进行分组，您甚至可以将 DVC 用于您的 git/bucket/gdrive，以避免 git 存储库因大文件而饱和。 这样做你不需要创建全局变量，只需实例化模型，你可以随时生成预测。

如果您有任何疑问，请随时问我。

链接到我的项目：https://github.com/Nouvellie/django-mlops-docker/blob/main/src/main/apps/mlops/utils/model_loader.py