<Neck_2>change Concat to WeightedAdd
This commit is contained in:
1 parent
e10e5dd65a
commit
8241503580
4 files changed
+69
-20
No files matched your search
@@ -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])
|
||||
|
||||
Reference in new issue
Block a user