Rename is_strided with _is_cn.

Author: datumbox

torchvision/models/detection/backbone_utils.py

@@ -138,7 +138,7 @@ def mobilenet_backbone(
 
     # Gather the indeces of blocks which are strided. These are the locations of C1, ..., Cn-1 blocks.
     # The first and last blocks are always included because they are the C0 (conv1) and Cn.
-    stage_indices = [0] + [i for i, b in enumerate(backbone) if getattr(b, "is_strided", False)] + [len(backbone) - 1]
+    stage_indices = [0] + [i for i, b in enumerate(backbone) if getattr(b, "_is_cn", False)] + [len(backbone) - 1]
     num_stages = len(stage_indices)
 
     # find the index of the layer from which we wont freeze

torchvision/models/mobilenetv2.py

@@ -88,7 +88,7 @@ def __init__(
         ])
         self.conv = nn.Sequential(*layers)
         self.out_channels = oup
-        self.is_strided = stride > 1
+        self._is_cn = stride > 1
 
     def forward(self, x: Tensor) -> Tensor:
         if self.use_res_connect:

torchvision/models/mobilenetv3.py

@@ -82,7 +82,7 @@ def __init__(self, cnf: InvertedResidualConfig, norm_layer: Callable[..., nn.Mod
 
         self.block = nn.Sequential(*layers)
         self.out_channels = cnf.out_channels
-        self.is_strided = cnf.stride > 1
+        self._is_cn = cnf.stride > 1
 
     def forward(self, input: Tensor) -> Tensor:
         result = self.block(input)

torchvision/models/segmentation/segmentation.py

@@ -34,8 +34,7 @@ def _segm_model(name, backbone_name, num_classes, aux, pretrained_backbone=True)
 
         # Gather the indeces of blocks which are strided. These are the locations of C1, ..., Cn-1 blocks.
         # The first and last blocks are always included because they are the C0 (conv1) and Cn.
-        stage_indices = [0] + [i for i, b in enumerate(backbone) if getattr(b, "is_strided", False)] + [
-            len(backbone) - 1]
+        stage_indices = [0] + [i for i, b in enumerate(backbone) if getattr(b, "_is_cn", False)] + [len(backbone) - 1]
         out_pos = stage_indices[-1]
         out_layer = str(out_pos)
         out_inplanes = backbone[out_pos].out_channels