tiny_yolov4.zip

上传者: Twilight737 | 上传时间: 2021-11-24 12:08:47 | 文件大小: 174.15MB | 文件类型: ZIP
## 运行步骤 1. 运行voc_annotation.py,生成根目录下的2007_train.txt、2007_val.txt、2007_test.txt。 2. 运行kmeans_for_anchors.py,生成anchors.txt,也可以不运行这一步,直接使用model_data/yolo_anchors.txt,这是为YOLO 416*416 模型专门设计的,效果依旧很好。 3. 运行train.py,即可开始训练。 4. 运行predict.py进行检测,其中包含图片检测、文件夹遍历检测。 硬件环境:单张RTX8000显卡。 1. 对于训练阶段,batch=32,调用GPU每轮训练需要2min,而禁止GPU后每轮训练需要5min。 2. 对于推理阶段,VOC数据集调用GPU每帧0.03894278782533031s,禁止GPU后每帧0.06158580083645041s,真不愧是tiny_yolov4,在CPU上运行也能足够快。 而之前测试的yolov4模型调用GPU每帧0.17s,tiny_yolov4速度上比起yolov4足足快了5倍,但的确在VOC2007数据集上的检测精度不及,轮廓精度和分类类别精度均有欠缺。

文件下载

评论信息

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明