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

Posted 2023-02-23

【中文标题】在 Django 应用程序中加载 TensorFlow 模型的位置

【英文标题】：Where to load Tensorflow model in Django application

【发布时间】：2021-10-03 08:19:26

【问题描述】：

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

【问题讨论】：

你可以试试单例设计模式。

嗨！请看下面的文章。 medium.com/saarthi-ai/…

【参考方案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