mmtracking单目标跟踪数据集说明以及自定义数据集

官方数据集说明文档可参考:数据集准备 — MMTracking 0.14.0 文档

我对其单目标跟踪的数据集格式简单说明一下。

单目标跟踪用到的数据集包括:LaSOT、UAV123、TrackingNet、OTB100、GOT10k、VOT2018。单目标跟踪任务的训练和测试配置文件中,用到了 MSCOCO, ILSVRC, LaSOT, UAV123, TrackingNet, OTB100 和 GOT10k 数据集。

官方虽然提供了OTB100、VOT2018的数据集下载脚本,但下载速度非常慢,有的则官网都没了,所以想要下载OTB100、VOT2018、UAV123,最快的办法就是使用openxlab

以OTB100为例,查看网址数据集-OpenDataLab

下载方法:

pip install openxlab #安装

pip install -U openxlab #版本升级

openxlab login #进行登录,输入对应的AK/SK

openxlab dataset info --dataset-repo OpenDataLab/OTB100 #数据集信息及文件列表查看

openxlab dataset get --dataset-repo OpenDataLab/OTB100 #数据集下载

openxlab dataset download --dataset-repo OpenDataLab/OTB100 --source-path /raw --target-path data/ #数据集文件下载

数据集的结构,要按照官方文档1.5 数据集文件夹结构的说明放好,然后根据文档2. 转换标注格式 执行相应的脚本,生成统一的CocoVID格式,实际上就是在每个数据集下生成了一个annotations文件夹,里面放着一个txt,打开文档看一下:

格式为:图片序列路径、图片序列标注文件路径、开始帧、结束帧,开始帧默认就是1,结束帧可以理解为最后一张图片的序列,或者文件夹下图片的总数。

以data_seq/UAV123/bike1,anno/UAV123/bike1.txt,1,3085为例,data_seq/UAV123/bike1目录下放的是一个视频的图片序列

anno/UAV123/bike1.txt存放的是每张图片,同一个物体的矩形标注框位置703和361是矩形框左上角点的坐标,57是矩形宽度,114是矩形长度。

1为起始ID

3085为data_seq/UAV123/bike1下图片总数量。

了解了CocoVID的格式,那么自定义数据集就不是问题了。思路如下:

1、视频抽帧

视频文件抽帧,图片文件名从000001.jpg开始。

2、标注图片

建议一个视频只跟着一个运动的物体标注,如果数据集实在匮乏,也可以一个视频的图片序列标注多个运动物体,但只要每个运动的物体都生成独立的标注文件就可以了。

3、生成annotations

以“data_seq/UAV123/bike1,anno/UAV123/bike1.txt,1,3085”为例,一行一条数据生成annotations.txt即可。

  • 5
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
mmpose是一种基于Python语言和开源框架MMCV、MMDetection、MMTracking的开发工具,用于训练自己动物关键点数据集的模型。 首先,准备自己的动物关键点数据集。这个数据集需要包含大量的动物图片,每张图片都要标注出关键点的位置。关键点可以是动物身体的一些特定位置,例如头部、躯干、四肢等。标注时可以使用标注工具,将关键点位置用坐标进行标记。 接下来,需要根据数据集的格式修改配置文件。在mmpose中,可以使用YAML格式的配置文件来定义网络结构、训练参数等。可以根据自己的需求修改配置文件,例如设置网络的层数、特征提取方法、训练的batch size等。 然后,在训练之前,需要将数据集进行预处理和增强。预处理步骤可以包括图像的缩放、裁剪、翻转等操作,以及对关键点进行标准化处理。增强操作可以包括随机旋转、平移、亮度调整等,以扩充数据集的多样性。 接着,可以使用mmpose提供的训练命令来开始训练模型。根据配置文件的设置,模型会根据数据集进行多轮的训练,不断优化网络参数,使得网络能够准确地识别动物的关键点。 最后,在训练完成后,可以使用训练好的模型进行动物关键点检测。将新的动物图片输入到模型中,模型会输出这些图片中动物关键点的位置。可以利用这些关键点来进行动物姿态分析、行为识别等应用。 总之,mmpose提供了一套完整的训练自己动物关键点数据集的工具,通过一系列的步骤,可以训练出一个准确度较高的动物关键点检测模型,为动物行为研究和动物保护提供帮助。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值