mmaction2 指南 (3) 添加新的数据集

最新推荐文章于 2025-06-09 09:02:44 发布

Alex丶Chen

最新推荐文章于 2025-06-09 09:02:44 发布

阅读量1.5k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：代码随笔行为识别

本文链接：https://blog.csdn.net/njuptalex/article/details/113249959

本文档介绍了如何在mmaction2中添加新的数据集，包括将数据重新组织为已有的标注格式，如rawframe和video标注，以及自定义数据集的创建。通过继承BaseDataset并重写相应方法，可以实现在线和离线转换。同时，提供了自定义数据集的例子以及如何支持混合数据训练。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

mmaction2 指南

3. 添加新的数据集

自定义行为识别数据集

新的数据重新组织到已有格式

重新构建标注文件就行

标注格式一：
rawframe标注【路径，帧数，类别】

some/directory-1 163 1
some/directory-2 122 1
some/directory-3 258 2
some/directory-4 234 2
some/directory-5 295 3
some/directory-6 121 3

标注格式二：
video标注【路径、类别】

some/path/000.mp4 1
some/path/001.mp4 1
some/path/002.mp4 2
some/path/003.mp4 2
some/path/004.mp4 3
some/path/005.mp4 3

标注格式三：

json格式的标注

{
  "video1": {
      "duration_second": 211.53,
      "duration_frame": 6337,
      "annotations": [
          {
              "segment": [
                  30.025882995319815,
                  205.2318595943838
              ],
              "label": "Rock climbing"
          }
      ],
      "feature_frame": 6336,
      "fps": 30.0,
      "rfps": 29.9579255898
  },
  "video2": {
      "duration_second": 26.75,
      "duration_frame": 647,
      "annotations": [
          {
              "segment": [
                  2.578755070202808,
                  24.914101404056165
              ],
              "label": "Drinking beer"
          }
      ],
      "feature_frame": 624,
      "fps": 24.0,
      "rfps": 24.1869158879
  }
}

自定义数据集 class

online转换

继承 BaseDataset 重写 load_annotations(self), evaluate(self, results, metrics, logger) and dump_results(self, results, out) 可参考 RawframeDataset, VideoDataset or ActivityNetDataset.

offline转换
修改数据标注文件，让其符合RawframeDataset, VideoDataset or ActivityNetDataset的调用形式
例子：自己改下标注文件，调用RawframeDataset

...
# dataset settings
dataset_type = 'RawframeDataset'
data_root = 'path/to/your/root'
data_root_val = 'path/to/your/root_val'
ann_file_train = 'data/custom/custom_train_list.txt'