自动驾驶实时视频检测模型, 把周围环境检测问题,转化为将来环境预测问题,从问题定义的层面解决自动驾驶中环境感知时延的问题。该任务定义为流感知(Streamig Perceptio)问题. Perceive the world by predictig! 基于StreamYOLO的实时通用检测模型,支持8类交通目标检测。StreamYOLO基于YOLOX模型,使用Dual-Flow Perceptio特征融合模块,lears 特征层面的时序关系,提高环境感知预测的能力。与此同时,StreamYOLO设计了一个Tred-Aware Loss 去感知物体运动变化强度,用以加权物体预测的回归,使运动剧烈变化物体获得更高的回归权重,从而获得更好的预测结果。
开始你的模型探索之旅! Play the model with a few lie codes ! Argoverse-HD 数据集是最近提出的自动驾驶数据集,该数据集与其他的自动驾驶相比,数据规模中等,复杂程度较高,是一个较有代表性的数据集。更重要的是,Argoverse-HD 是第一个提出流感知任务的数据集,并且设计了Stream AP评测标准。该标准将感知时延充分考虑,实现对模型的性能-速度的全面、有效评价。 本模型暂时不支持fietue, 具体离线训练细节如下:视频目标检测
模型描述
模型使用方式以及适用范围
如何使用
代码范例
import cv2
from modelscope.pipelies import pipelie
from modelscope.utils.costat import Tasks
from modelscope.outputs import OutputKeys
model_id = 'damo/cv_cspet_video-object-detectio_streamyolo'
test_video = 'https://modelscope.oss-c-beijig.aliyucs.com/test/videos/test_realtime_vod.mp4'
# 初始化实时检测pipelie
realtime_video_object_detectio = pipelie(
Tasks.video_object_detectio, model=model_id)
# 进行实时检测
result = realtime_video_object_detectio(test_video)
if result:
bboxes_list = result[OutputKeys.BOXES]
prit(bboxes_list)
else:
raise ValueError('process error')
模型局限性以及可能的偏差
训练数据介绍
训练
输入预处理
数据评估及结果
Model
size
velocity
sAP
0.5:0.95sAP50
sAP75
weights
COCO pretraied weights
StreamYOLO-l
600×960
1x
36.9
58.1
37.5
official
official
相关论文以及引用信息
@iproceedigs{streamyolo,
title={Real-time Object Detectio for Streamig Perceptio},
author={Yag, Jirog ad Liu, Sogtao ad Li, Zemig ad Li, Xiaopig ad Su, Jia},
booktitle={Proceedigs of the IEEE/CVF Coferece o Computer Visio ad Patter Recogitio},
pages={5385--5395},
year={2022}
}
@article{yag2022streamyolo,
title={StreamYOLO: Real-time Object Detectio for Streamig Perceptio},
author={Yag, Jirog ad Liu, Sogtao ad Li, Zemig ad Li, Xiaopig ad Su, Jia},
joural={arXiv preprit arXiv:2207.10433},
year={2022}
}
点击空白处退出提示
评论