ebike_detect_2

我要开发同款
匿名用户2024年07月31日
53阅读

技术信息

开源地址
https://modelscope.cn/models/etoiles/ebike_detect_2
授权协议
Apache License 2.0

作品详情

模型描述

这里我们提供基于业界领先的目标检测框架DAMO-YOLO训练的检测模型:DAMO-YOLO-S。DAMO-YOLO是一个面向工业落地的目标检测框架,兼顾模型速度与精度,其训练的模型效果超越了目前的一众YOLO系列方法,并且仍然保持极高的推理速度。DAMO-YOLO引入TiyNAS技术,使得用户可以根据硬件算力进行低成本的检测模型定制,提高硬件利用效率并且获得更高精度。另外,DAMO-YOLO还对检测模型中的eck、head结构设计,以及训练时的标签分配、数据增广等关键因素进行了优化,具体细节可以参考我们的开源代码技术报告DAMO-YOLO-S是DAMO-YOLO提供的一系列模型中,平衡了速度和精度的最优模型之一。

模型评测

Model size mAPval
0.5:0.95
Latecy(ms)
T4-TRT-FP16
FLOPs
(G)
Parameters(M)
YOLOX-S 640 40.5 3.20 26.8 9.0
YOLOv5-S 640 37.4 3.04 16.5 7.2
YOLOv6-S 640 43.5 3.10 44.2 17.0
PP-YOLOE-S 640 43.0 3.21 17.4 7.9
DAMO-YOLO-S 640 46.8 3.83 37.8 16.3
  • 表中汇报的mAP是COCO2017 val集上的结果。
  • 表中汇报的latecy不包括后处理(ms)时间,其测试条件为:T4 GPU,TesorRT=7.2.1.6, CUDA=10.2, CUDNN=8.0.0.1。

使用范围

本模型适用范围较广,能对图片中包含的大部分前景物体(COCO 80类)进行定位。

使用方法

在ModelScope框架上,提供输入图片,即可以通过简单的Pipelie调用使用当前模型。具体代码示例如下:

from modelscope.pipelies import pipelie
from modelscope.utils.costat import Tasks
object_detect = pipelie(Tasks.image_object_detectio,model='damo/cv_tiyas_object-detectio_damoyolo')
img_path ='https://modelscope.oss-c-beijig.aliyucs.com/test/images/image_detectio.jpg'
result = object_detect(img_path)

模型fietue

DAMO-YOLO现已支持模型fietue,要使用modelscope进行fietue需要首先将您的自有数据的图片和标注准换成标准COCO格式,fietue的样例如下。

    # 准备预训练模型
    model_id = 'damo/cv_tiyas_object-detectio_damoyolo'
    cache_path = sapshot_dowload(model_id)
    # 开始训练
    kwargs = dict(
            cfg_file=os.path.joi(cache_path, 'cofiguratio.jso'),
            gpu_ids=[
                0,
            ],
            batch_size=16,
            max_epochs=3,
            um_classes=1,
            pretrai_model=os.path.joi(cache_path,
                                        'damoyolo_tiyasL25_S.pt'),
            cache_path=cache_path,
            trai_image_dir='./data/test/images/image_detectio/images',
            val_image_dir='./data/test/images/image_detectio/images',
            trai_a=
            './data/test/images/image_detectio/aotatios/coco_sample.jso',
            val_a=
            './data/test/images/image_detectio/aotatios/coco_sample.jso')
        traier = build_traier(
            ame=Traiers.tiyas_damoyolo, default_args=kwargs)
        traier.trai()

模型可视化效果

引用

 @article{damoyolo,
  title={DAMO-YOLO: A Report o Real-Time Object Detectio Desig},
  author={Xiazhe Xu, Yiqi Jiag, Weihua Che, Yilu Huag, Yua Zhag ad Xiuyu Su},
  joural={arXiv preprit arXiv:2211.15444v2},
  year={2022}
}

功能介绍

模型描述 这里我们提供基于业界领先的目标检测框架DAMO-YOLO训练的检测模型:DAMO-YOLO-S。DAMO-YOLO是一个面向工业落地的目标检测框架,兼顾模型速度与精度,其训练的模型效果超越了

声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论