"docker/ascend_npu/requirements.txt" did not exist on "a296ea41f9d14cda6257eca51ce089531029a2ba"
__init__.py 1022 Bytes
Newer Older
1
2
3
4
5
from .accuracy import Accuracy, accuracy
from .balanced_l1_loss import BalancedL1Loss, balanced_l1_loss
from .cross_entropy_loss import (CrossEntropyLoss, binary_cross_entropy,
                                 cross_entropy, mask_cross_entropy)
from .focal_loss import FocalLoss, sigmoid_focal_loss
6
from .ghm_loss import GHMC, GHMR
7
8
from .iou_loss import (BoundedIoULoss, GIoULoss, IoULoss, bounded_iou_loss,
                       iou_loss)
9
10
from .mse_loss import MSELoss, mse_loss
from .smooth_l1_loss import SmoothL1Loss, smooth_l1_loss
Kai Chen's avatar
Kai Chen committed
11
from .utils import reduce_loss, weight_reduce_loss, weighted_loss
Jiangmiao Pang's avatar
Jiangmiao Pang committed
12

13
__all__ = [
Kai Chen's avatar
Kai Chen committed
14
15
16
    'accuracy', 'Accuracy', 'cross_entropy', 'binary_cross_entropy',
    'mask_cross_entropy', 'CrossEntropyLoss', 'sigmoid_focal_loss',
    'FocalLoss', 'smooth_l1_loss', 'SmoothL1Loss', 'balanced_l1_loss',
17
    'BalancedL1Loss', 'mse_loss', 'MSELoss', 'iou_loss', 'bounded_iou_loss',
18
    'IoULoss', 'BoundedIoULoss', 'GIoULoss', 'GHMC', 'GHMR', 'reduce_loss',
19
    'weight_reduce_loss', 'weighted_loss'
20
]