(Deeplabv3+MobilenetV2)语义分割模型部署手机端(ckpt-pb-tflite)

最近要做一个实时的语义分割项目,需要完成手机端的部署。这篇文章主要是对模型转换做一个总结。 首先模型训练,不多说,网上有很多资源,但是tf官网保存的模型是ckpt格式,要完成移动端部署,需要将ckpt转换成freeze pb,再将pb转换成tflite。 1.ckpt转换成pb /research/deeplab目录下有一个export_model.py文件,这个文件可以直接将ckpt转换成pb,
相关文章
相关标签/搜索