training/loss/__init__.py

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)