API обнаружения объектов TensorFlow и модуль утилит

Question

Я только начинаю с TensorFlow и натолкнулся на учебник по API обнаружения объектов TensorFlow 1002 *. Я выполнил шаги по установке, описанные в первом разделе, создал новую виртуальную среду conda (в Visual Studio 2017) и установил TensorFlow, используя pip. Также я установил пакеты, перечисленные в других разделах.

Это импорт, взятый отсюда: Обнаружение объектов с помощью веб-камеры

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'

Так что же это за модуль и где его взять?

Answer 1

Вы также можете установить API обнаружения объектов в вашу среду python / conda, используя

python setup.py build
python setup.py install

Это обеспечит автоматический поиск пакетов в вашей среде conda. Файл setup.py находится в папке models / research.

Answer 2

Добавьте корневой каталог API обнаружения объектов (...\models\research\object_detection) в PYTHONPATH:

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