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

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

tf2运行YOLOv4(tflite)相关推荐

睿智的目标检测32——TF2搭建YoloV4目标检测平台（tensorflow2）
睿智的目标检测32--TF2搭建YoloV4目标检测平台(tensorflow2) 学习前言什么是YOLOV4 代码下载 YOLOV4改进的部分(不完全) YOLOV4结构解析 1.主干特征提取网络 ...
vs2017运行yolov4_VS2017运行YOLOv4
YOLO依赖opencv,所以需要在VS2017中配置opencv.YOLO使用的是COCO数据集,感兴趣的可以移步官网. [配置opencv]属性表方式新建opencv3410_release_x ...
win10运行YOLOv4+OPENCV+VS2017
方案一.QtGUI+YOLOv4+OPENCV+VS2017 主要参考:https://blog.csdn.net/weixin_42448226/article/details/105752224 ...
yolov4 开发环境搭建_YOLOv4 的各种新实现、配置、测试、训练资源汇总
点击上方"计算机视觉life",选择"星标" 快速获得最新干货近日最火的莫过于 YOLOv4 的横空出世,CV君在第一时间进行了 YOLOv4的论文解读: 得 ...
YOLOV4+DeepSort车流量检测基础版
YOLOV4+DeepSort车流量检测基础版资源下载: 链接:https://pan.baidu.com/s/1Mv1VI1Wa5MUaHuKz9hlniQ 提取码:lmpk 复制这段内容后打开百 ...
yolov4网络结构_重磅更新！YoLov4最新论文！解读YoLov4框架!
早上刷到YOLOv4之时,非常不敢相信这是真的!今明两天各大公众号肯定要炸锅了! 论文:https://arxiv.org/pdf/2004.10934.pdf 源码:https://github.c ...
yolov4论文_YOLOv4论文详细解读
大家好,本次分享的是YOLOv4的那篇论文我将主要从这几个部分对该论文进行详细的解读首先来看论文的主要内容YOLOv4这篇论文主要是介绍如何使用一些通用的新方法达到了最先进的实验结果,他们在COCO数 ...
YOLOv4论文笔记
论文:YOLOv4: Optimal Speed and Accuracy of Object Detection.Alexey Bochkovskiy,Chien-Yao Wang,Hong-Yua ...
YOLOv4论文(中文版)
摘要据说有大量的特征可以提高卷积神经网络(CNN)的准确性.需要在大数据集上对这些特征的组合进行实际测试,并对结果进行理论验证.有些特征专门针对某些模型和某些问题,或者只针对小规模数据集;而一些特性 ...

tf2运行YOLOv4(tflite)

tf2运行YOLOv4(tflite)相关推荐

最新文章

热门文章