(Deeplabv3+MobilenetV2)語義分割模型部署手機端(ckpt-pb-tflite)

最近要做一個實時的語義分割項目,需要完成手機端的部署。這篇文章主要是對模型轉換做一個總結。 首先模型訓練,不多說,網上有很多資源,但是tf官網保存的模型是ckpt格式,要完成移動端部署,需要將ckpt轉換成freeze pb,再將pb轉換成tflite。 1.ckpt轉換成pb /research/deeplab目錄下有一個export_model.py文件,這個文件可以直接將ckpt轉換成pb,
相關文章
相關標籤/搜索