slowfast训练自己的数据集

Question

Traceback (most recent call last):
    train_loader = loader.construct_loader(cfg, "train")
  File "d:\file\postgrad\experiment\code\slowfast-main\slowfast\datasets\loader.py", line 115, in construct_loader
    dataset = build_dataset(dataset_name, cfg, split)
  File "d:\file\postgrad\experiment\code\slowfast-main\slowfast\datasets\build.py", line 31, in build_dataset
    return DATASET_REGISTRY.get(name)(cfg, split)
  File "d:\file\postgrad\experiment\code\slowfast-main\slowfast\datasets\ava_dataset.py", line 50, in __init__
    self._load_data(cfg)
  File "d:\file\postgrad\experiment\code\slowfast-main\slowfast\datasets\ava_dataset.py", line 66, in _load_data
    boxes_and_labels = ava_helper.load_boxes_and_labels(cfg, mode=self._split)
  File "d:\file\postgrad\experiment\code\slowfast-main\slowfast\datasets\ava_helper.py", line 93, in load_boxes_and_labels
    all_boxes, count, unique_box_count = parse_bboxes_file(
  File "d:\file\postgrad\experiment\code\slowfast-main\slowfast\datasets\ava_helper.py", line 208, in parse_bboxes_file
    if box_key not in all_boxes[video_name][frame_sec]:
KeyError: 1

大家在用slowfast自制数据集有遇到过这个报错嘛，想问问怎么解决的，非常感谢

Answer 1

帅哥，望采纳一下,我拿个牌不容易啊
检查数据集文件：

确认数据集中的文件是否符合预期格式，特别是与 ava_helper.py 中的 parse_bboxes_file 相关的部分。确保所有需要的键（例如 video_name 和 frame_sec）在数据文件中都有定义。
调试数据：

在 parse_bboxes_file 函数中，添加调试信息（如打印语句），查看 all_boxes 的结构，确认是否 video_name 和 frame_sec 的值匹配。
确保数据完整性：

验证数据加载过程中的每一步，确保没有数据丢失或格式不一致的情况。例如，检查是否所有视频和帧都有对应的框（box）和标签（label）。
修改代码：

如果数据不完整，修改代码以处理缺失的键。例如，使用 dict.get(key, default) 方法避免直接访问不存在的键。

if box_key not in all_boxes.get(video_name, {}).get(frame_sec, {}):
    # 处理缺失的情况

更新数据集处理脚本：

检查 ava_dataset.py 和 ava_helper.py 中是否有硬编码的值或者假设数据格式的地方。确保这些部分能够处理实际数据的多样性和异常情况。
检查数据加载配置：

确保配置文件中指定的路径、文件名和参数与实际数据一致。
咨询文档和社区：

查看 SlowFast 的文档和相关社区，可能会发现类似的问题和解决方案。