如何解决运行 export_inference_graph.py 抛出值错误
我是 StackOverflow 的新手。我正在尝试使用 export_inference_graph.py 导出模型。 我使用faster_rcnn_inception_v2在本地训练了我的模型。我正在关注这个tutorial。
在命令提示符下输入
python export_inference_graph.py --input_type image_tensor --pipeline_config_path CAPTCHA_training/faster_rcnn_inception_v2_coco.config --trained_checkpoint_prefix "CAPTCHA_training_dir/model.ckpt-51272" --output_directory CAPTCHA_inference_graph
所有路径正确的我都会收到以下错误。
File "export_inference_graph.py",line 206,in <module>
tf.app.run()
File "C:\Users\Jatin\anaconda3\lib\site-packages\tensorflow\python\platform\app.py",line 40,in run
_run(main=main,argv=argv,flags_parser=_parse_flags_tolerate_undef)
File "C:\Users\Jatin\anaconda3\lib\site-packages\absl\app.py",line 303,in run
_run_main(main,args)
File "C:\Users\Jatin\anaconda3\lib\site-packages\absl\app.py",line 251,in _run_main
sys.exit(main(argv))
File "export_inference_graph.py",line 194,in main
exporter.export_inference_graph(
File "C:\Users\Jatin\anaconda3\lib\site-packages\object_detection\exporter.py",line 604,in export_inference_graph
detection_model = model_builder.build(pipeline_config.model,File "C:\Users\Jatin\anaconda3\lib\site-packages\object_detection\builders\model_builder.py",line 1116,in build
return build_func(getattr(model_config,meta_architecture),is_training,line 583,in _build_faster_rcnn_model
_check_feature_extractor_exists(frcnn_config.feature_extractor.type)
File "C:\Users\Jatin\anaconda3\lib\site-packages\object_detection\builders\model_builder.py",line 249,in _check_feature_extractor_exists
raise ValueError('{} is not supported. See `model_builder.py` for features '
ValueError: faster_rcnn_inception_v2 is not supported. See `model_builder.py` for features extractors compatible with different versions of Tensorflow
我使用的是 Python 3.8.5 和 tensorflow 版本 2.4.1 提前致谢
解决方法
根据错误看起来像一个Tensorflow版本问题。查看源 faster_rcnn_inception_v2
存在于 if tf_version.is_tf1():
下。尝试使用 TF 1。
faster_rcnn_inception_v2_coco
存在于 Tensorflow 1 检测模型动物园中。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。