tensorflow检测API SsdFeatureExtractor“对象没有属性”“override”“base”“feature”“extractor”“超参数”

2024-03-29 05:06:43 发布

您现在位置:Python中文网/ 问答频道 /正文

当我使用ssd_mobilenet_v1_coco_11_06_2017模型来训练自己的数据集时

我使用googletesorflow检测api来训练它

会出现一些问题。 我的操作系统:ubuntu 16.04

./train.sh
Traceback (most recent call last): File "../../train.py", line 167, in tf.app.run() File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/platform/app.py", line 126, in run _sys.exit(main(argv)) File "../../train.py", line 163, in main worker_job_name, is_chief, FLAGS.train_dir) File "/home/feng/project/models/research/object_detection/trainer.py", line 240, in train detection_model = create_model_fn() File "/home/feng/project/models/research/object_detection/builders/model_builder.py", line 98, in build add_background_class) File "/home/feng/project/models/research/object_detection/builders/model_builder.py", line 166, in _build_ssd_model is_training=is_training) File "/home/feng/project/models/research/object_detection/builders/model_builder.py", line 129, in _build_ssd_feature_extractor feature_extractor_config.override_base_feature_extractor_hyperparams)

AttributeError: 'SsdFeatureExtractor' object has no attribute 'override_base_feature_extractor_hyperparams'

这是什么问题?


Tags: inpyprojecthomemodelobjectmodelsline
2条回答

我最近升级了TensorFlow对象检测API并通过刷新protobuf绑定修复了这个问题。在

之所以会出现这种错误消息,是因为安装了错误的protobuf绑定,因为新绑定可能与仍在引用的旧版本绑定不兼容。您可以通过下载protobuf的最新版本并编译新的绑定来解决这个问题。在

要执行此操作,请遵循此处的“手动编译器安装和使用方法”的说明:Section Link,但是为了将来针对将来断开的链接进行证明,我将注意,他们当前指示您:

  1. 使tensorflow/models/research成为当前目录
  2. 下载并安装最新版本的protoc(又名protobuf)
    • wget -O protobuf.ziP
    • https://github.com/google/protobuf/releases/download/v3.0.0/protoc-3.0.0-linux-x86_64.zip
    • unzip protobuf.zip
  3. 使用下载的版本运行编译过程(旧版本可能仍在您的路径中,您可能在其他地方需要它)。
    • ./bin/protoc object_detection/protos/*.proto python_out=.
  4. 将库添加到PYTHONPATH
    • export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim
  5. 测试安装
    • python object_detection/builders/model_builder_test.py

我建议您查看TensorFlow对象检测API Git报告的以下问题:

https://github.com/tensorflow/models/issues/4121

相关问题 更多 >