TensorFlow 对象检测 API 和实用程序模块

Question

我刚开始使用 TensorFlow，遇到了 TensorFlow Object Detection API tutorial。我按照第一部分中概述的安装步骤创建了一个新的 conda 虚拟环境（在 Visual Studio 2017 内）并使用 pip 安装了 TensorFlow。我还安装了其他部分中列出的软件包。

这是从这里导入的：Detect Objects Using Your Webcam

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 的 package/module。毫不奇怪，尝试导入它失败了：

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

这是什么模块，从哪里可以得到？

Answer 1

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

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

Answer 2

您还可以使用

将对象检测 api 安装到您的 python/conda 环境中

python setup.py build
python setup.py install

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

TensorFlow 对象检测 API 和实用程序模块

TensorFlow Object Detection API and utils module

python

visual-studio-2012

tensorflow