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

不過我運行時發(fā)現(xiàn)一個bug，就是在OpenCV畫框時，浮點型會報錯，所以就修復了一下，工程為：https://github.com/moneypi/tensorflow-yolov4-tflite

?

要注意的幾點，一個是該工程是建議使用?Tensorflow 2.3，確實如此，如果用Tensorflow 2.4行不行？好像是不行，貌似TensorFlow官方搞了一個bug，導致運行時會報錯。

所以建議版本為Tensorflow 2.3，CPU版本即可，反而是GPU版本，如果是檢測可能可以，不過在轉(zhuǎn)格式的時候，如果GPU顯存不足，反而會崩潰

?

因為我是想在開發(fā)板(jetson)上運行，發(fā)現(xiàn)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幀的樣子

所以，想要尋找一個折中的神經(jīng)網(wǎng)絡，還在嘗試中，既能保證精度，又能保證速度

總結

以上是生活随笔為你收集整理的tf2运行YOLOv4(tflite)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。