tensorflow - 尝试使用张量流 2 对象检测 API 运行训练时,获取此 ImportErrorcan "not import name fpn_pb2"
问题描述
我正在使用 Tensor flow 2 对象检测 API 进行深度学习研究。运行模型训练时出现此错误。我按照 Gilbert Tanner 和 Edje Electronics 教程进行了基本安装和环境设置。我正在使用 TensorFlow 对象检测 API 新的 GitHub Commit。我将所有 .protos 文件转换为 .py 文件,但仍面临此错误。我附上了这个错误的截图。请检查一下,如果您能提供帮助,请告诉我。
提前致谢。错误
解决方案
我也遇到了同样的问题target_assigner.proto
,center_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