[Fix] Fix import errors in custom CD models

own
Bobholamovic 3 years ago
parent 1d92bb5f10
commit ba4d0d09a6
  1. 4
      paddlers/custom_models/cd/__init__.py
  2. 2
      paddlers/custom_models/cd/changestar.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

@ -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

Loading…
Cancel
Save