自制AVA数据集工具/slowfast模型训练数据集制作

AVA2.1版本更新了追踪这一列。

那2.2版本请参考此链接文章: 自定义ava数据集及训练与测试 完整版 时空动作/行为 视频数据集制作 yolov5, deep sort, VIA MMAction, SlowFast_CV-杨帆的博客-CSDN博客_ava数据集训练

本文提供了 AVA V2.1 版本数据集制作脚本。

V2.2版本多了一列追踪。

觉得有用的可以顺手在github点个star,给你们比心心,哈哈哈

提供了 train/val.csv v2.1文件生成,

如果你做slowfast行为检测,还提供 annotations\person_box_67091280_iou90下文件即包含prediecred框文件的生成。

如果你使用的facebook(meta)的slowfast,本脚本完全适用。

如果你用的openmmlab平台里的sowfast,需要自行对含有score的csv文件做一个转pkl的脚本,并且删除action kind一列。

自动获取time,坐标,自动获取score。

action kind 手动在 VIA3.0 上标注。

详细信息请阅读README

GitHub – Blueyao17/self-made_ava_dataset_tool: Some scripts on generating homemade AVA format datasets Some scripts on generating homemade AVA format datasets – GitHub – Blueyao17/self-made_ava_dataset_tool: Some scripts on generating homemade AVA format datasets 自制AVA数据集工具/slowfast模型训练数据集制作https://github.com/Blueyao17/self-made_ava_dataset_tool ;上述脚本可能还有一些不完善的地方,希望同行们继续完善修改。

train.csv V2.1

自制AVA数据集工具/slowfast模型训练数据集制作

annotations\person_box_67091280_iou90 下的文件:(slowfast模型AVA训练需要)

自制AVA数据集工具/slowfast模型训练数据集制作

Original: https://blog.csdn.net/Bluemoon17/article/details/123238573
Author: 曼彻斯特的蓝
Title: 自制AVA数据集工具/slowfast模型训练数据集制作

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/532559/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球