Create model/backbones/mobilenetv2.py

model/backbones/mobilenetv2.py

@@ -0,0 +1,18 @@
+import torch.nn as nn
+from torchvision.models import mobilenet_v2
+
+
+class MobileNetV2Backbone(nn.Module):
+    def __init__(self):
+        super().__init__()
+        model = mobilenet_v2(pretrained=True)
+        self.model = model.features
+        self.enc_channels = [24, 32, 96, 320]  # фиксированные выходы энкодера
+
+    def forward(self, x):
+        features = []
+        for idx, layer in enumerate(self.model):
+            x = layer(x)
+            if idx in {2, 4, 7, 13}:
+                features.append(x)
+        return features