Skip to content

Commit d62060e

Browse files
committed
Revert branch5 of aspp
1 parent 4fa75c7 commit d62060e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

models/proposed.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -38,13 +38,13 @@ def __init__(self, in_channels: int, out_channels: int):
3838
# 5번 branch = AdaptiveAvgPool2d → 1x1 convolution → BatchNorm → ReLu
3939
self.branch5 = nn.Sequential(
4040
nn.AdaptiveAvgPool2d(1),
41-
nn.Conv2d(in_channels, in_channels // 2, kernel_size=1),
42-
nn.BatchNorm2d(in_channels // 2),
41+
nn.Conv2d(in_channels, out_channels, kernel_size=1),
42+
nn.BatchNorm2d(out_channels),
4343
nn.ReLU(inplace=True)
4444
)
4545
# 최종 출력 convolution
4646
self.outconv = nn.Sequential(
47-
nn.Conv2d(out_channels * 4 + in_channels // 2, out_channels, kernel_size=1),
47+
nn.Conv2d(out_channels * 5, out_channels, kernel_size=1),
4848
nn.BatchNorm2d(out_channels),
4949
nn.ReLU(inplace=True)
5050
)

0 commit comments

Comments
 (0)