- 在👉链接处将代码库下载下来
!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 --weight weights/yolov5s.pt --cfg models/yolov5s.yaml --img 320 --tfl-int8 --source 你的数据集路径 --ncalib 图像数量
转化为int8的需要数据集来进行验证,所以需要添加自己数据集的路径,图像数量为你所用的用于验证的图像数。
转化完成后有如下结果: