仅下载 TensorFlow 库的特定部分
Posted
技术标签:
【中文标题】仅下载 TensorFlow 库的特定部分【英文标题】：Download only specific part of Tensorflow Library 【发布时间】：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 个函数，而不是整个库...感谢期待。
【问题讨论】：
【参考方案1】：虽然我对整个库不太确定，但有一个精简版的 Tensorflow（我猜他们意识到 430MB 也有点多）。
可以在此处找到有关此的信息： https://www.tensorflow.org/lite/
这里的指南似乎详细介绍了如何挑选和选择 Lite 库的部分，虽然我自己没有使用过，但我应该期望两者之间有一定程度的兼容性......
https://www.tensorflow.org/lite/guide/reduce_binary_size
【讨论】：
以上是关于仅下载 TensorFlow 库的特定部分的主要内容，如果未能解决你的问题，请参考以下文章