فهرست منبع

Fix changestar

Bobholamovic 2 سال پیش
والد
کامیت
95a0f9e2ac
1فایلهای تغییر یافته به همراه4 افزوده شده و 1 حذف شده
  1. 4 1
      paddlers/rs_models/cd/changestar.py

+ 4 - 1
paddlers/rs_models/cd/changestar.py

@@ -121,7 +121,10 @@ class ChangeStar_FarSeg(_ChangeStarBase):
             def forward(self, x):
                 return self._seg_model(x)
 
-        seg_model = FarSeg(decoder_out_channels=mid_channels)
+        seg_model = FarSeg(
+            in_channels=3,
+            num_classes=num_classes,
+            decoder_out_channels=mid_channels)
 
         super(ChangeStar_FarSeg, self).__init__(
             seg_model=_FarSegWrapper(seg_model),