EasyCV/easycv/models/loss/__init__.py
2022-08-25 16:57:37 +08:00

8 lines
311 B
Python

# Copyright (c) Alibaba, Inc. and its affiliates.
from .cross_entropy_loss import CrossEntropyLoss
from .face_keypoint_loss import FacePoseLoss, WingLossWithPose
from .focal_loss import FocalLoss
from .iou_loss import GIoULoss, IoULoss
from .mse_loss import JointsMSELoss
from .pytorch_metric_learning import *