首页 > 其他 > 详细

YOLOV3——PyTorch训练TensorFlowLite部署模型转换

时间:2020-02-13 14:30:22      阅读:631      评论:0      收藏:0      [点我收藏+]
  • 主要思路 

          将训练好的.pt文件转换为keras的.h5文件,再将.h5文件转换为.tflite文件。

 

  • 步骤:

       1.环境:PyTorch1.0以上(其他版本也可以,主要是适配下面的开源代码)

       .pt转weights下载后在文件内添加pt_to_weights.py。

from models import *

model = Darknet("cfg/yolov3-obj.cfg")#你用于训练的cfg文件
load_darknet_weights(model,"weights/latest.pt")
save_weights(model,path=weights/latest.weights,cutoff=-1)

      2.环境:Tensorflow2.0,要安装keras模块

      .weights转.h5

      修改convert.py中

          技术分享图片

        需要避免[None,None,None,3],设置为input_layer = Input(shape=(229, 229, 3)),tensorflow中为静态图,输入要确定。

        运行

  python convert.py yolov3-obj.cfg latest.weights latest.h5

        3.环境:TensorFlow2.0

import tensorflow as tf
converter = tf.lite.TFLiteConverter.from_keras_model_file(latest.h5)
tflite_model = converter.convert()
open("latest.tflite", "wb").write(tflite_model)

         生成后验证是否正确识别即可

YOLOV3——PyTorch训练TensorFlowLite部署模型转换

原文:https://www.cnblogs.com/thunder-wu/p/12303460.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!