2024年Python最新【教程】YOLOv5模型转化-Android端部署_yolov5 android

那时我开始思考，官方明明白白地写了是PyTorch > ONNX > CoreML > TFLite，为什么最后没有.tflite模型文件呢，难不成上当了？其实没有，大佬zldrobit已经写出了转化代码，其pull request👉链接，该仓库的代码根据YOLOv5的发展也在改进，唯一不同的是增加了tflite转化的代码

博主在colab平台进行如下操作，具体使用方法如下：

• 在👉链接处将代码库下载下来

!git clone https://github.com/zldrobit/yolov5.git
%cd yolov5
!git checkout tf-android # 或者tf-only-export

• 安装所需要的库

!pip install -r requirements.txt
!pip install tensorflow==2.4.0

• 运行yolov5/models路径下的tf.py代码即可

!python models/tf.py --weights weights/yolov5s.pt --cfg models/yolov5s.yaml --img 320

这样就能得到fp16的.tflite模型，若需要Int8的，则用如下命令

!python models/tf.py