TensorFlow对象检测API和utils模块

时间:2019-01-17 21:53:24

标签: python tensorflow visual-studio-2012

我只是从TensorFlow开始,遇到了substr()。我已按照第一部分中概述的安装步骤进行操作,创建了一个新的conda虚拟环境(在Visual Studio 2017中),并使用TensorFlow安装了pip。另外,我还安装了其他部分列出的软件包。

这是从此处获取的进口:TensorFlow Object Detection API tutorial

import numpy as np
import os
import six.moves.urllib as urllib
import sys
import tarfile
import tensorflow as tf
import zipfile
import cv2

from collections import defaultdict
from io import StringIO
from matplotlib import pyplot as plt
from PIL import Image
from utils import label_map_util
from utils import visualization_utils as vis_util

但是找不到名为utils的程序包/模块。毫不奇怪,尝试导入失败会失败:

>>> from utils import label_map_util
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'utils'

那这是什么模块,我从哪里可以得到它?

2 个答案:

答案 0 :(得分:1)

通过以下方式将对象检测API(...\models\research\object_detection)的根目录添加到PYTHONPATH

export PYTHONPATH=\path\to\models\research\object_detection\:$PYTHONPATH

答案 1 :(得分:0)

您还可以使用

将对象检测api安装到python / conda环境中。
python setup.py build
python setup.py install

这将确保您的conda环境自动找到软件包。 setup.py文件在models / research文件夹中。