NFL Health & Safety - Helmet Assignment 是Kaggle上的一个热门竞赛,由于其高额的奖金和极富挑战的任务设置吸引了许多实力强劲的队伍参与竞争。NFL Helmet检测在其中具有十分关键的地位,在此,我们提供基于DAMO-YOLO-S的NFL Helmet检测模型,您可以将其作为您的基础模型之一加入到方案中,实现更高的精度,欢迎试用。
Model | size | mAPval 0.5:0.95 |
Latency(ms) T4-TRT-FP16 |
FLOPs (G) |
Parameters(M) |
---|---|---|---|---|---|
DAMO-YOLO-S | 640 | – | 3.83 | 37.8 | 16.3 |
本模型适用于NFL Health & Safety - Helmet Assignment竞赛,能对图片中的NFL Helmet进行定位。
在ModelScope框架上,提供输入图片,即可以通过简单的Pipeline调用使用当前模型。具体代码示例如下:
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
object_detect = pipeline(Tasks.image_object_detection,model='XianzheXu/cv_tinynas_object-detection_damoyolo_nfl-helmet')
img_path ='./assets/nfl_1.jpg'
result = object_detect(img_path)
数据来源于NFL Health & Safety - Helmet Assignment赛事官方,数据集中包含一个类别,即NFL-Helmet。
@article{damoyolo,
title={DAMO-YOLO: A Report on Real-Time Object Detection Design},
author={Xianzhe Xu, Yiqi Jiang, Weihua Chen, Yilun Huang, Yuan Zhang and Xiuyu Sun},
journal={arXiv preprint arXiv:2211.15444v2},
year={2022}
}