Update model/backbones/mobilenetv2.py

model/backbones/mobilenetv2.py

@@ -1,20 +1,12 @@
 import torch.nn as nn
-from torchvision.models import mobilenet_v2
-
-
-
+from torchvision.models import mobilenet_v2, MobileNet_V2_Weights
 
 class MobileNetV2Backbone(nn.Module):
     def __init__(self):
-        super().__init__()
-        model = mobilenet_v2(pretrained=True)
-        self.model = backbone    #model.features
-        self.enc_channels = [24, 32, 96, 160, 320]  # фиксированные выходы энкодера
+        super(MobileNetV2Backbone, self).__init__()
+        # Используем актуальный способ загрузки предобученной модели
+        self.model = mobilenet_v2(weights=MobileNet_V2_Weights.DEFAULT).features
 
     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
+        return self.model(x)
+