#将ckpt文件转为.pb文件
freeze_graph \
--input_graph=./****/graph_eval.pbtxt \
--input_checkpoint=./****/model.ckpt-238818 \
--output_graph=./****/frozen_eval_graph-238818-20000.pb \
--output_node_names=head/reg13x13_output/BiasAdd,head/reg26x26/BiasAdd,head
echo "freeze graph done." #输出节点name,我的是yolov3,因此是三个输出name
#生成移动端部署的tflite
tflite_convert \
--output_file=./*****/float-mbV1yoloV3.tflite \
--graph_def_file=./*****/frozen_eval_graph-245400-20000.pb \
--input_arrays=Placeholder \
--output_arrays=head/reg13x13_output/BiasAdd,head/reg26x26/BiasAdd,head/reg52x52/BiasAdd
echo "tflite convertion done."
tensorflow将.ckpt转为.pb文件,将.pb文件转为.tflite文件
最新推荐文章于 2023-07-31 22:00:00 发布