我知道大多数从事tensorflow对象检测工作的人都遇到过这个问题。我的tensorflow项目在另一个文件夹中运行,但现在我确实喜欢使用模型文件夹中存储的对象检测功能。
我已经将python脚本中的工作目录更改为模型/对象检测文件夹,方法如下:
import sys
import os
changed_directory = False
if not changed_directory:
os.chdir(os.path.join('../models/research/','object_detection/'))
changed_directory = True
然后,我在脚本中以这种方式附加路径:
import sys
sys.path.append("...models\research\object_detection\")
这就是我的模型文件夹看起来像的样子
TensorFlow/
NAR_object_detection/
└─ models/
├─ community/
├─ official/
├─ orbit/
├─ research/
└── ...
研究模型中的对象检测文件夹。同样重要的是,我的对象检测python脚本存储在Tensorflow中名为NAR_object_detection的文件夹中,所以我的意思是NAR_object_detection文件夹与 models文件夹当我尝试以这种方式导入时,我得到了一个错误,即没有称为对象检测的模块 为什么import语句找不到对象检测文件夹?from object_detection.utils import label_map_util
尝试这个
pip install tensorflow-object-detection-api