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帧的样子

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