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

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

?

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

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

?

因?yàn)槲沂窍朐陂_發(fā)板(jetson)上運(yùn)行，發(fā)現(xiàn)416的分辨率，運(yùn)行還是很慢，所以改為320的分辨率，運(yùn)行腳本如下：

# 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

說句實(shí)在話，也挺卡的，感覺每秒2、3幀的樣子

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

總結(jié)

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

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