From 17f68aeec269cae9c5bdcaa892cc0149e09bc39a Mon Sep 17 00:00:00 2001 From: Yanyi Liu Date: Tue, 11 Oct 2022 11:45:14 +0800 Subject: [PATCH] fix channel --- mmrotate/models/dense_heads/s2a_head.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmrotate/models/dense_heads/s2a_head.py b/mmrotate/models/dense_heads/s2a_head.py index 36303caa1..ec47b6efe 100644 --- a/mmrotate/models/dense_heads/s2a_head.py +++ b/mmrotate/models/dense_heads/s2a_head.py @@ -87,12 +87,12 @@ def __init__(self, def _init_layers(self) -> None: """Initialize layers of the head.""" self.or_conv = ORConv2d( - self.feat_channels, + self.in_channels, int(self.feat_channels / 8), kernel_size=3, padding=1, arf_config=(1, 8)) - self.or_pool = RotationInvariantPooling(256, 8) + self.or_pool = RotationInvariantPooling(self.feat_channels, 8) self.relu = nn.ReLU(inplace=True) self.cls_convs = nn.ModuleList() self.reg_convs = nn.ModuleList()