其他分享
首页 > 其他分享> > tf2运行YOLOv4(tflite)

tf2运行YOLOv4(tflite)

作者:互联网

原工程为 https://github.com/hunglc007/tensorflow-yolov4-tflite

不过我运行时发现一个bug,就是在OpenCV画框时,浮点型会报错,所以就修复了一下,工程为:https://github.com/moneypi/tensorflow-yolov4-tflite

 

要注意的几点,一个是该工程是建议使用 Tensorflow 2.3,确实如此,如果用Tensorflow 2.4行不行?好像是不行,貌似TensorFlow官方搞了一个bug,导致运行时会报错。

所以建议版本为Tensorflow 2.3,CPU版本即可,反而是GPU版本,如果是检测可能可以,不过在转格式的时候,如果GPU显存不足,反而会崩溃

 

因为我是想在开发板(jetson)上运行,发现416的分辨率,运行还是很慢,所以改为320的分辨率,运行脚本如下:

# Save tf model for tflite converting
python save_model.py --weights ./data/yolov4-tiny.weights --output ./checkpoints/yolov4-tiny-320 --input_size 320 --model yolov4 --framework tflite --tiny

# yolov4 tiny
python convert_tflite.py --weights ./checkpoints/yolov4-tiny-320 --output ./checkpoints/yolov4-tiny-320.tflite

检测脚本为:

python3 detect.py --weights ./checkpoints/yolov4-tiny-320.tflite --size 320 --model yolov4 --image ./data/kite.jpg --framework tflite

说句实在话,也挺卡的,感觉每秒2、3帧的样子

所以,想要寻找一个折中的神经网络,还在尝试中,既能保证精度,又能保证速度

标签:YOLOv4,tflite,tiny,--,320,yolov4,checkpoints,tf2
来源: https://blog.csdn.net/zhqh100/article/details/117527355