瀏覽代碼

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),