yolov5模型剪枝报错：KeyError: 'model.0.bn'

Question

问题遇到的现象和发生背景

使用https://gitcode.net/mirrors/midasklr/yolov5prune/-/tree/v6.0

运行结果及报错内容

Traceback (most recent call last):

  File "D:\1-syf\yolov5\prune.py", line 812, in <module>
    main(opt)

  File "D:\1-syf\yolov5\prune.py", line 785, in main
    run_prune(**vars(opt))

  File "C:\ProgramData\Anaconda3\lib\site-packages\torch\autograd\grad_mode.py", line 27, in decorate_context
    return func(*args, **kwargs)

  File "D:\1-syf\yolov5\prune.py", line 491, in run_prune
    pruned_model = ModelPruned(maskbndict=maskbndict, cfg=pruned_yaml, ch=3).to(device)

  File "D:\1-syf\yolov5\models\yolo.py", line 263, in __init__
    self.model, self.save, self.from_to_map = parse_pruned_model(self.maskbndict, deepcopy(self.yaml), ch=[ch])  # model, savelist

  File "D:\1-syf\yolov5\models\yolo.py", line 477, in parse_pruned_model
    bnc = int(maskbndict[named_m_bn].sum())

KeyError: 'model.0.bn'

我的解答思路和尝试过的方法

未搜索到解决办法。

Answer 1

把maskbndict打印出来看看