Tensorflow对象检测API中的错误

Question

我正在关注this tutorial，

但是当我运行命令时

python generate_tfrecord.py --csv_input=images\train_labels.csv --image_dir=images\train --output_path=train.record

我收到以下错误，我该如何解决问题

C:\Users\ASUS PC\Anaconda3\lib\site-packages\h5py\__init__.py:36: FutureWarning: Conversion of the second argument of issubdtype from `float` to `np.floating` is deprecated. In future, it will be treated as `np.float64 == np.dtype(float).type`.
  from ._conv import register_converters as _register_converters
Traceback (most recent call last):
  File "generate_tfrecord.py", line 20, in <module>
    from object_detection.utils import dataset_util
ModuleNotFoundError: No module named 'object_detection'

Answer 1

您可能忘记这样做，或根据您自己的安装更改路径：

set PYTHONPATH=C:\tensorflow1\models;C:\tensorflow1\models\research;C:\tensorflow1\models\research\slim

您应该将C:\tensorflow1\models替换为计算机上tensorflow / models目录的路径。

或者您已退出虚拟环境并返回虚拟环境，因此您需要再次使用此命令。 （参见链接教程的第2e节）