SlowFast报错：ValueError: too many values to unpack (expected 4)

SlowFast报错：ValueError: too many values to unpack (expected 4)

报错细节

  File "/home/user/yuanjinmin/SlowFast/tools/visualization.py", line 81, in run_visualization
    for inputs, labels, _, meta in tqdm.tqdm(vis_loader):
ValueError: too many values to unpack (expected 4)

报错原因

cfg.TENSORBOARD.MODEL_VIS设置为True，此时会执行SlowFast/tools/visualization.py下第83行的代码：

for inputs, labels, _, meta in tqdm.tqdm(vis_loader):

但vis_loader中每个可迭代的对象是

可以加个

first_batch = next(iter(vis_loader))
print(first_batch)

进行验证

debug流程及修改

debug方法可参考之前的博文

我是在测试SlowFast模型时遇到的问题，而cfg.TENSORBOARD.MODEL_VIS是TENSORBOARD的可视化，这个是测试完最后执行的，测试大概需要4-5个小时的时间（单卡3090），因此为了尽快debug需要复制run_net.py为visualization_net.py，并修改为：

#!/usr/bin/env python3

"""可视化模型"""
from slowfast.config.defaults import assert_and_infer_cfg
from slowfast.utils.misc import launch_job
from slowfast.utils.parser import load_config, parse_args

from demo_net import demo
from test_net import test
from train_net import train
from visualization import visualize


def main():
    """
    Main function to spawn the train and test process.
    """
    # todo
    import pydevd_pycharm

    pydevd_pycharm.settrace('localhost', port=29501, stdoutToServer=True, stderrToServer=True)

    args = parse_args()
    print("config files: {}".format(args.cfg_files))
    for path_to_config in args.cfg_files:
        cfg = load_config(args, path_to_config)
        cfg = assert_and_infer_cfg(cfg)

        # Perform model visualization.
        if cfg.TENSORBOARD.ENABLE and (
                cfg.TENSORBOARD.MODEL_VIS.ENABLE
                or cfg.TENSORBOARD.WRONG_PRED_VIS.ENABLE
        ):
            launch_job(cfg=cfg, init_method=args.init_method, func=visualize)


if __name__ == "__main__":
    main()

在此基础上debugSlowFast/tools/visualization.py下第83行的代码，可以看到：

每个可迭代的对象有5个成员，因此在这里多加一个_,即可

    # todo 此处增加了一个_，因为vis_loader的返回值是5个，而原来的是4个
    # for inputs, labels, _, meta in tqdm.tqdm(vis_loader):
    for inputs, labels, _, _, meta in tqdm.tqdm(vis_loader):

当然，也可直接将cfg.TENSORBOARD.MODEL_VIS设置为False，建议这个还是不要打开，非常地耗时…