4 个月前

YOLOv7:可训练的免费技巧集在实时目标检测器中创造了新的最先进水平

YOLOv7:可训练的免费技巧集在实时目标检测器中创造了新的最先进水平

摘要

YOLOv7 在从 5 FPS 到 160 FPS 的速度范围内,超越了所有已知的目标检测器,在 GPU V100 上以 30 FPS 或更高的帧率运行时,其精度达到了所有已知实时目标检测器中的最高值 56.8% AP。YOLOv7-E6 目标检测器(在 V100 上达到 56 FPS,精度为 55.9% AP)在速度上比基于变压器的检测器 SWIN-L Cascade-Mask R-CNN(在 A100 上达到 9.2 FPS,精度为 53.9% AP)快 509%,精度高 2%;同时在速度上比基于卷积的检测器 ConvNeXt-XL Cascade-Mask R-CNN(在 A100 上达到 8.6 FPS,精度为 55.2% AP)快 551%,精度高 0.7% AP。此外,YOLOv7 还在速度和精度方面超越了其他多种目标检测器,包括 YOLOR、YOLOX、Scaled-YOLOv4、YOLOv5、DETR、Deformable DETR、DINO-5scale-R50、ViT-Adapter-B 等。更为重要的是,我们仅使用 MS COCO 数据集从零开始训练 YOLOv7,未使用任何其他数据集或预训练权重。源代码已发布在 https://github.com/WongKinYiu/yolov7。

代码仓库

AlexeyAB/darknet
tf
GitHub 中提及
WongKinYiu/YOLO
pytorch
GitHub 中提及
mkang315/rcs-yolo
pytorch
GitHub 中提及
ogatarina-sq/secyolov7
pytorch
GitHub 中提及
xinwei666/mmgenerativeir
pytorch
GitHub 中提及
pjreddie/darknet
pytorch
GitHub 中提及
wongkinyiu/yolov7
官方
pytorch
GitHub 中提及
LdDl/rust-road-traffic
GitHub 中提及
zhanghuiyao/yolov7_mindspore
mindspore
GitHub 中提及
mikel-brostrom/yolov7
pytorch
GitHub 中提及
henrytsui000/YOLO
pytorch
GitHub 中提及

基准测试

基准方法指标
2d-object-detection-on-ceymoYOLOv7
mAP: 69.5
object-detection-on-cocoYOLOv7-D6 (44 fps)
box mAP: 56.6
object-detection-on-cocoYOLOv7-E6 (56 fps)
box mAP: 56
object-detection-on-cocoYOLOv7 (161 fps)
box mAP: 51.4
object-detection-on-cocoYOLOv7-X (114 fps)
box mAP: 53.1
object-detection-on-cocoYOLOv7-W6 (84 fps)
box mAP: 54.9
object-detection-on-coco-oYOLOv7-E6E
Average mAP: 32.0
Effective Robustness: 6.42
pedestrian-detection-on-dvtodYOLOv7 (Visible)
mAP: 35.3
pedestrian-detection-on-dvtodYOLOv7 (Thermal)
mAP: 77.8
real-time-object-detection-on-cocoYOLOv7-X
FPS (V100, b=1): 114
box AP: 53.1
real-time-object-detection-on-cocoYOLOv7-D6(1280)
FPS (V100, b=1): 44
box AP: 56.6
real-time-object-detection-on-cocoYOLOv7-E6E(1280)
FPS (V100, b=1): 36
box AP: 56.8
real-time-object-detection-on-cocoYOLOv7-W6(1280)
FPS (V100, b=1): 84
box AP: 54.9
real-time-object-detection-on-cocoYOLOv7-E6(1280)
FPS (V100, b=1): 56
box AP: 56

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
YOLOv7:可训练的免费技巧集在实时目标检测器中创造了新的最先进水平 | 论文 | HyperAI超神经