如何检查哪个Tensorflow版本与Tensorflow模型园兼容？
提问于 2022-02-08 02:49:04
为了使用带有Tensorflow的预训练模型，我们克隆TensorFlow模型园，然后在模型动物园中选择一个模型，例如，检测模型动物园：EfficientDet D0 512x512。
是否可以检测到Tensorflow的正确版本，例如2.7.0、2.7.1或2.8.0，这样肯定会与上述设置一起工作吗？
文档(README.md)似乎没有提到这一要求。也许这是暗示的？
我检查了setup.py的对象检测，但仍然没有线索！
\models\research\object_detection\packages\tf2\setup.pyREQUIRED_PACKAGES = [
# Required for apache-beam with PY3
'avro-python3',
'apache-beam',
'pillow',
'lxml',
'matplotlib',
'Cython',
'contextlib2',
'tf-slim',
'six',
'pycocotools',
'lvis',
'scipy',
'pandas',
'tf-models-official>=2.5.1',
'tensorflow_io',
'keras'
]
回答 1
Stack Overflow用户
发布于 2022-02-08 08:20:24
我不知道有一种正式/快速的方法来确定的右Tensorflow版本，给出一个特定的 模范花园版本master分支。然而，这里是我的解决办法：
- 在上面的
REQUIRED_PACKAGES中，我们看到了tf-models-official>=2.5.1。 - 检查包装历史 on pypi.org，最新版本，截止到03.02.2022，是
2.8.0。 - 因此，在安装这个
\models\research\object_detection\packages\tf2\setup.py文件时，pip自然会获取tf-models-official的最新版本，即2.8.0，这要归功于>=符号。 - 但是，给定
tf-models-official、v2.8.0，其所需的包是在tf-models-official-2.8.0\tf_models_official.egg-info\requires.txt中定义的(注意:下载包并使用链接解压缩)。 - 在这里，我们发现：
tensorflow~=2.8.0...meaning所需的Tensorflow版本是2.8.*。
这可能是不需要的，例如在CoLab中，目前的版本是2.7.0。
- 为了解决这个问题，我们应该使用
tf-models-official，v2.7.0。注意，它与Tensorflow版本匹配。在这个版本的2.7.0的requires.txt中，我们应该看到tensorflow>=2.4.0，它已经被CoLab的默认Tensorflow版本(2.7.0)所满足。 - 为了解决这一问题，应该将
\models\research\object_detection\packages\tf2\setup.py从'tf-models-official>=2.5.1'修改为'tf-models-official==2.7.0'。
警告：我认为这一攻击不会影响对象检测API的功能，因为它最初需要任何tf-models-official >= 2.5.1。我们只需将其修改为==2.7.0即可。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/71032585
复制相关文章
腾讯云开发者
