【Tensorflow踩过的坑儿】pb转pbtxt

本文详细介绍了如何使用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配置文件

评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值