Explorar o código

[Fix] Fix import errors in custom CD models

Bobholamovic %!s(int64=3) %!d(string=hai) anos
pai
achega
ba4d0d09a6

+ 2 - 2
paddlers/custom_models/cd/__init__.py

@@ -20,5 +20,5 @@ from .snunet import SNUNet
 from .dsamnet import DSAMNet
 from .changestar import ChangeStar
 from .fc_ef import FCEarlyFusion
-from .fc_siamconc import FCSiamConc
-from .fc_siamdiff import FCSiamDiff
+from .fc_siam_conc import FCSiamConc
+from .fc_siam_diff import FCSiamDiff

+ 1 - 1
paddlers/custom_models/cd/changestar.py

@@ -17,7 +17,7 @@ import paddle.nn as nn
 import paddle.nn.functional as F
 
 from paddlers.datasets.cd_dataset import MaskType
-from paddlers.custom_models.seg.models import FarSeg
+from paddlers.custom_models.seg import FarSeg
 from .layers import Conv3x3, Identity