bluspater commited on
Commit
3b09131
·
verified ·
1 Parent(s): 782b77c

Update model/backbones/mobilenetv2.py

Browse files
Files changed (1) hide show
  1. model/backbones/mobilenetv2.py +1 -1
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, 320]
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)