【发布时间】：2022-01-09 12:36:52
【问题描述】：
我有一个用于对象检测的深度学习代码。我所做的是在 Google Colab 上运行代码，然后导出模型以在本地使用它。现在要运行模型，我必须再次安装整个 Tensorflow 包，这对我的系统来说非常重。
我想问一下有没有办法只下载和运行 Tensorflow 库的特定部分？
我只在代码中的 2 个地方使用了 Tensorflow，我必须为其安装整个 Tensorflow 库。
-
这是我加载模型的地方。
detect_fn = tf.saved_model.load(PATH_TO_SAVED_MODEL) -
这是我第二次使用 Tensorflow 的地方。
input_tensor = tf.convert_to_tensor(image_rgb)
这些是 Tensorflow 库中我需要的仅有的 2 个函数，而不是整个库...感谢期待。
【问题讨论】：
标签： python tensorflow tensorflow2.0 tensor tensorflow-datasets