【Tensorflow踩过的坑儿】pb转pbtxt

最新推荐文章于 2021-03-09 11:56:46 发布

原创最新推荐文章于 2021-03-09 11:56:46 发布 · 7k 阅读

6 ·

CC 4.0 BY-SA版权

机器学习专栏收录该内容

1 篇文章

订阅专栏

本文详细介绍了如何使用tf_text_graph_mask_rcnn.py脚本来转换TensorFlow的Mask RCNN模型。通过修改该脚本中的参数，包括输入的.pb模型路径、输出的.pbtxt文本图模型路径以及配置文件路径，可以实现模型的有效转换。

修改tf_text_graph_mask_rcnn.py文件，下载链接

parser.add_argument('--input', default='F:/tensorflow_object_detection/object_detection_12_29/pd_model/mask_rcnn_inception_v2_coco/frozen_inference_graph.pb', help='Path to frozen TensorFlow graph.')
parser.add_argument('--output', default='F:/tensorflow_object_detection/object_detection_12_29/pd_model/mask_rcnn_inception_v2_coco/mask_rcnn.pbtxt', help='Path to output text graph.')
parser.add_argument('--config', default='F:/tensorflow_object_detection/object_detection_12_29/samples/configs/mask_rcnn_inception_v2_coco.config', help='Path to a *.config file is used for training.')
args = parser.parse_args()

其中：input是转换后的.pb模型，output是输出的.pbtxt文件，config是mask_rcnn_inception_v2_coco.config配置文件