model/backbones.py

import torch.nn as nn
from torchvision.models import mobilenet_v2, MobileNet_V2_Weights

class MobileNetV2Backbone(nn.Module):
    def __init__(self):
        super(MobileNetV2Backbone, self).__init__()
        # Используем актуальный способ загрузки предобученной модели
        self.model = mobilenet_v2(weights=MobileNet_V2_Weights.DEFAULT).features
        self.enc_channels = [24, 32, 96, 320]  # Каналы с разных уровней feature extractor

    def forward(self, x):
        return self.model(x)