Skip to content

Commit 899b78c

Browse files
committed
🐛 [AAAI|Fix] put reconbranch for c model
1 parent 2a68c65 commit 899b78c

1 file changed

Lines changed: 25 additions & 1 deletion

File tree

yolo/config/model/aaai-c.yaml

Lines changed: 25 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,31 @@ model:
5353
args: {out_channels: 256, part_channels: 256}
5454
tags: N3
5555

56-
reconstruct: []
56+
reconstruct:
57+
- UpSample:
58+
args: {scale_factor: 2, mode: nearest}
59+
- RepConv:
60+
args: {out_channels: 256}
61+
- RepConv:
62+
args: {out_channels: 128, activation: false}
63+
tags: R2
64+
65+
- UpSample:
66+
args: {scale_factor: 2, mode: nearest}
67+
- RepConv:
68+
args: {out_channels: 128}
69+
- RepConv:
70+
args: {out_channels: 64, activation: false}
71+
tags: R1
72+
73+
- UpSample:
74+
args: {scale_factor: 2, mode: nearest}
75+
- RepConv:
76+
args: {out_channels: 64}
77+
- RepConv:
78+
args: {out_channels: 3, activation: false}
79+
tags: RMAP
80+
output: True
5781

5882
head:
5983
- ADown:

0 commit comments

Comments
 (0)