首页 > 解决方案 > 尝试使用张量流 2 对象检测 API 运行训练时,获取此 ImportErrorcan "not import name fpn_pb2"

问题描述

我正在使用 Tensor flow 2 对象检测 API 进行深度学习研究。运行模型训练时出现此错误。我按照 Gilbert Tanner 和 Edje Electronics 教程进行了基本安装和环境设置。我正在使用 TensorFlow 对象检测 API 新的 GitHub Commit。我将所有 .protos 文件转换为 .py 文件,但仍面临此错误。我附上了这个错误的截图。请检查一下,如果您能提供帮助,请告诉我。

提前致谢。错误

标签: tensorflowtensorflow2.0object-detection-api

解决方案


我也遇到了同样的问题target_assigner.protocenter_net.proto你必须将这三个都添加.\object_detection\protos\fpn.proto .\object_detection\protos\target_assigner.proto .\object_detection\protos\center_net.proto 到 protoc 命令中。所以整个命令应该是:

protoc --python_out=. .\object_detection\protos\anchor_generator.proto .\object_detection\protos\argmax_matcher.proto .\object_detection\protos\bipartite_matcher.proto .\object_detection\protos\box_coder.proto .\object_detection\protos\box_predictor.proto .\object_detection\protos\eval.proto .\object_detection\protos\faster_rcnn.proto .\object_detection\protos\faster_rcnn_box_coder.proto .\object_detection\protos\grid_anchor_generator.proto .\object_detection\protos\hyperparams.proto .\object_detection\protos\image_resizer.proto .\object_detection\protos\input_reader.proto .\object_detection\protos\losses.proto .\object_detection\protos\matcher.proto .\object_detection\protos\mean_stddev_box_coder.proto .\object_detection\protos\model.proto .\object_detection\protos\optimizer.proto .\object_detection\protos\pipeline.proto .\object_detection\protos\post_processing.proto .\object_detection\protos\preprocessor.proto .\object_detection\protos\region_similarity_calculator.proto .\object_detection\protos\square_box_coder.proto .\object_detection\protos\ssd.proto .\object_detection\protos\fpn.proto .\object_detection\protos\target_assigner.proto .\object_detection\protos\center_net.proto .\object_detection\protos\ssd_anchor_generator.proto .\object_detection\protos\string_int_label_map.proto .\object_detection\protos\train.proto .\object_detection\protos\keypoint_box_coder.proto .\object_detection\protos\multiscale_anchor_generator.proto .\object_detection\protos\graph_rewriter.proto .\object_detection\protos\calibration.proto .\object_detection\protos\flexible_grid_anchor_generator.proto

推荐阅读