这里我们提供基于业界领先的目标检测框架 本模型适用范围较广,能对图片中包含的大部分前景物体(COCO 80类)进行定位。 在ModelScope框架上,提供输入图片,即可以通过简单的Pipelie调用使用当前模型。具体代码示例如下: DAMO-YOLO现已支持模型fietue,要使用modelscope进行fietue需要首先将您的自有数据的图片和标注准换成标准COCO格式,fietue的样例如下。模型描述
模型评测
Model
size
mAPval
0.5:0.95Latecy(ms)
T4-TRT-FP16FLOPs
(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
640
46.8
3.83
37.8
16.3
使用范围
使用方法
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
# 准备预训练模型
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}
}
点击空白处退出提示
评论