File size: 1,299 Bytes
8bc3305 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | import os
import sys
current_file_path = os.path.abspath(__file__)
parent_dir = os.path.dirname(os.path.dirname(current_file_path))
project_root_dir = os.path.dirname(parent_dir)
sys.path.append(parent_dir)
sys.path.append(project_root_dir)
from metrics.registry import LOSSFUNC
from .cross_entropy_loss import CrossEntropyLoss
from .consistency_loss import ConsistencyCos
from .capsule_loss import CapsuleLoss
from .bce_loss import BCELoss
from .am_softmax import AMSoftmaxLoss
from .am_softmax import AMSoftmax_OHEM
from .contrastive_regularization import ContrastiveLoss
from .l1_loss import L1Loss
from .id_loss import IDLoss
from .vgg_loss import VGGLoss
from .js_loss import JS_Loss
from .patch_consistency_loss import PatchConsistencyLoss
from .region_independent_loss import RegionIndependentLoss
from .supercontrast_loss import SupConLoss
from .supercontrast_cls_loss import SupConClsLoss
from .cross_entropy_orth_loss import CrossEntropyOrthLoss
from .cross_entropy_orth1_loss import CrossEntropyOrth1Loss # CE + Orth + MSE
from .cross_entropy_orth2_loss import CrossEntropyOrth2Loss # CE + Orth + CosineSim
from .cross_entropy_orth3_loss import CrossEntropyOrth3Loss # CE + + MSE
from .cross_entropy_orth4_loss import CrossEntropyOrth4Loss # CE + Orth + RelaxedCosineSim(th=0.9)
|