对象检测安装测试需要tensorflow_hub吗?

时间:2020-04-27 16:51:24

标签: tensorflow object-detection tensorflow-hub

我尝试根据此处说明的手册安装tensorflow对象检测模型: https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/install.html

基于已定义的步骤,我安装了tensorflow 1.14,使用tensorflow模型R1.13,python3.7和 一切正常,直到使用object_detection.ipynb最终测试安装为止。

我遇到以下错误:

ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-5-f127d2b71be1> in <module>
      4 import tensorflow as tf
      5 
----> 6 import tensorflow_hub as hub
      7 
      8 # For downloading the image.

ModuleNotFoundError: No module named 'tensorflow_hub' 

当我尝试安装 tensorflow_hub 时,它要求tf> 1.17,并且此安装会全部升级 库。从另一个角度来看,当我将最后一个库用于张量流时,由于tf2之间的不兼容,无法重新训练模块。和tf1。

有什么想法如何安装对象检测模块以使用自定义数据集进行重新训练?

0 个答案:

没有答案