Spaces:
Runtime error
Runtime error
Update model/backbones/mobilenetv2.py
Browse files
model/backbones/mobilenetv2.py
CHANGED
|
@@ -13,7 +13,7 @@ class MobileNetV2Backbone(nn.Module):
|
|
| 13 |
self.enc8x = features[7:14] # Примерно до блока 13
|
| 14 |
self.enc32x = features[14:] # Оставшиеся блоки
|
| 15 |
|
| 16 |
-
self.enc_channels = [24, 32, 96,
|
| 17 |
|
| 18 |
def forward(self, x):
|
| 19 |
out2x = self.enc2x(x)
|
|
|
|
| 13 |
self.enc8x = features[7:14] # Примерно до блока 13
|
| 14 |
self.enc32x = features[14:] # Оставшиеся блоки
|
| 15 |
|
| 16 |
+
self.enc_channels = [24, 32, 96, 1280] # Обновлено: финальный слой имеет 1280 каналов
|
| 17 |
|
| 18 |
def forward(self, x):
|
| 19 |
out2x = self.enc2x(x)
|