search

OpenGVLab / UniFormerV2

[ICCV2023] UniFormerV2: Spatiotemporal Learning by Arming Image ViTs with Video UniFormer
https://arxiv.org/abs/2211.09552
Apache License 2.0
291 stars 18 forks source link

您好 关于在自己的数据集上进行训练 #72

Closed ZachWong123 closed 4 months ago

ZachWong123 commented 4 months ago

由于我的数据集中的动作有和k400重合的部分 所以我保留了k400原本的映射关系 并且加入了我新加的50种动作的的映射关系 按照这个格式生成了新的.csv文件 123 于是我将config.yaml中的num_classes设置为451 并且将数据集路径改为我的视频和.csv文件存放的路径 但是在我启动任务的时候 会卡在 123123 start epoch这一步 我debug代码 发现代码运行到这一行 好像就进入循环了 123132 我不知道现在要如何解决这个问题 恳切希望可以得到您的答复

Andy1621 commented 4 months ago

可以到dataset的__getitem__打印看看

ZachWong123 commented 4 months ago

非常感谢您的提示 我使用少量视频对比发现 在k400上跑的时候 运行到 Q%G`Q6M$O_@Z3OOHI KF5F0 _getitem_可以正确print读取到的视频 但是使用我自己构建的数据 这边没有办法print出来

ZachWong123 commented 4 months ago

IMG_0684 可以正确获取到我自己构建数据的路径以及labels

ZachWong123 commented 4 months ago

L$PRZ~Z)__{X8)C_Z%J)5G 并且不跑train 单纯使用我构建的数据 跑测试 也能正确调用getitem

ZachWong123 commented 4 months ago

问题已经解决 非常感谢您的回复