检测backbone添加out_channels属性，方便后续module获取

Author: WenmuZhou

torchocr/networks/backbones/DetMobilenetV3.py

@@ -184,8 +184,10 @@ def __init__(self, in_channels, **kwargs):
         inplanes = self.make_divisible(inplanes * scale)
         self.stages = nn.ModuleList()
         block_list = []
+        self.out_channels = []
         for layer_cfg in cfg:
             if layer_cfg[5] == 2 and i > 2:
+                self.out_channels.append(inplanes)
                 self.stages.append(nn.Sequential(*block_list))
                 block_list = []
             block = ResidualUnit(num_in_filter=inplanes,
@@ -207,6 +209,7 @@ def __init__(self, in_channels, **kwargs):
             padding=0,
             groups=1,
             act='hard_swish')
+        self.out_channels.append(self.make_divisible(scale * cls_ch_squeeze))
 
     def make_divisible(self, v, divisor=8, min_value=None):
         if min_value is None:
@@ -235,4 +238,4 @@ def forward(self, x):
             x = stage(x)
             out.append(x)
         out[-1] = self.conv2(out[-1])
-        return out
+        return out

torchocr/networks/backbones/DetResNetvd.py

@@ -209,6 +209,7 @@ def __init__(self, in_channels, layers, **kwargs):
         self.pool1 = nn.MaxPool2d(kernel_size=3, stride=2, padding=1)
 
         self.stages = nn.ModuleList()
+        self.out_channels = []
         in_ch = 64
         for block_index in range(len(depth)):
             block_list = []
@@ -227,8 +228,8 @@ def __init__(self, in_channels, layers, **kwargs):
                                               stride=2 if i == 0 and block_index != 0 else 1,
                                               if_first=block_index == i == 0, name=conv_name))
                 in_ch = block_list[-1].output_channels
+            self.out_channels.append(in_ch)
             self.stages.append(nn.Sequential(*block_list))
-        self.out_channels = in_ch
 
     def load_3rd_state_dict(self, _3rd_name, _state):
         if _3rd_name == 'paddle':