seg剪枝怎么改啊还是是这样

[yinheyu]

Search before asking

• I have searched the YOLOv5 issues and discussions and found no similar questions.

Question

   elif m is Segment:
        for j in range(3):
            ignore_idx.append(named_m_base + ".m.{}".format(j))
            from_to_map[named_m_base + ".m.{}".format(j)] = fromlayer[f[j]]
        for j in range(3):
            ignore_idx.append(named_m_base + ".proto.cv{}.conv".format(j))
            fromlayer.append(named_m_base + ".proto.cv{}.conv".format(j))

报错：
Traceback (most recent call last):
File "D:\YOLOv5-Light-main\pruning\slim.py", line 115, in
prune_and_export(model, ignore_idx, opt)
File "D:\YOLOv5-Light-main\pruning\slim.py", line 53, in prune_and_export
weights_inheritance(model, compact_model, from_to_map, maskbndict)
File "D:\YOLOv5-Light-main\utils\prune_utils.py", line 198, in weights_inheritance
former = from_to_map[layername]
KeyError: 'model.24.proto.cv1.conv'

Additional

No response

[yinheyu]

elif m is Segment:
          for j in range(3):
              ignore_idx.append(named_m_base + ".proto.cv{}.conv".format(j))
              from_to_map[named_m_base + ".proto.cv{}.conv".format(j)] = fromlayer[f[j]]

Traceback (most recent call last):
File "D:\YOLOv5-Light-main\pruning\slim.py", line 115, in
prune_and_export(model, ignore_idx, opt)
File "D:\YOLOv5-Light-main\pruning\slim.py", line 53, in prune_and_export
weights_inheritance(model, compact_model, from_to_map, maskbndict)
File "D:\YOLOv5-Light-main\utils\prune_utils.py", line 198, in weights_inheritance
former = from_to_map[layername]
KeyError: 'model.24.m.0'

[github-actions]

👋 Hello, this issue has been automatically marked as stale because it has not had recent activity. Please note it will be closed if no further activity occurs.

Access additional YOLOv5 🚀 resources:

• Wiki – https://github.com/ultralytics/yolov5/wiki
• Tutorials – https://github.com/ultralytics/yolov5#tutorials
• Docs – https://docs.ultralytics.com

Access additional Ultralytics ⚡ resources:

• Ultralytics HUB – https://ultralytics.com/hub
• Vision API – https://ultralytics.com/yolov5
• About Us – https://ultralytics.com/about
• Join Our Team – https://ultralytics.com/work
• Contact Us – https://ultralytics.com/contact

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐!