<Neck_2>change Concat to WeightedAdd

This commit is contained in:
Zhang Wei committed 2025-12-24 16:29:39 +08:00
1 parent e10e5dd65a
commit 8241503580
4 files changed
+69 -20

No files matched your search

+6
View File
@@ -37,6 +37,7 @@ from models.common import (
C3x,
Classify,
Concat,
WeightedAdd, # ✅ 新增
Contract,
Conv,
CrossConv,
@@ -433,6 +434,11 @@ def parse_model(d, ch):
args = [ch[f]]
elif m is Concat:
c2 = sum(ch[x] for x in f)
elif m is WeightedAdd:
# 加权相加:所有输入通道必须一致
c2 = ch[f[0]]
# TODO: channel, gw, gd
elif m in {Detect, Segment}:
args.append([ch[x] for x in f])