当我使用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'
这是什么问题?
我最近升级了TensorFlow对象检测API并通过刷新protobuf绑定修复了这个问题。在
之所以会出现这种错误消息,是因为安装了错误的protobuf绑定,因为新绑定可能与仍在引用的旧版本绑定不兼容。您可以通过下载protobuf的最新版本并编译新的绑定来解决这个问题。在
要执行此操作,请遵循此处的“手动编译器安装和使用方法”的说明:Section Link,但是为了将来针对将来断开的链接进行证明,我将注意,他们当前指示您:
tensorflow/models/research
成为当前目录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
./bin/protoc object_detection/protos/*.proto python_out=.
PYTHONPATH
export PYTHONPATH=$PYTHONPATH:
`pwd`
:
`pwd`
/slim
python object_detection/builders/model_builder_test.py
我建议您查看TensorFlow对象检测API Git报告的以下问题:
https://github.com/tensorflow/models/issues/4121
相关问题 更多 >
编程相关推荐