Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 380 Bytes
Newer Older
wuyuefeng's avatar
semantic head with unittest  
wuyuefeng committed
1 
from mmdet.models.losses import FocalLoss, SmoothL1Loss, binary_cross_entropy
Yezhen Cong's avatar
Added axis-aligned IoU loss to VoteNet (#194)  
Yezhen Cong committed
2 
from .axis_aligned_iou_loss import AxisAlignedIoULoss, axis_aligned_iou_loss
wuyuefeng's avatar
Votenet  
wuyuefeng committed
3 
from .chamfer_distance import ChamferDistance, chamfer_distance
zhangwenwei's avatar
Initial commit  
zhangwenwei committed
4
wuyuefeng's avatar
Votenet  
wuyuefeng committed
5
6 
__all__ = [
    'FocalLoss', 'SmoothL1Loss', 'binary_cross_entropy', 'ChamferDistance',
Yezhen Cong's avatar
Added axis-aligned IoU loss to VoteNet (#194)  
Yezhen Cong committed
7 
    'chamfer_distance', 'axis_aligned_iou_loss', 'AxisAlignedIoULoss'
wuyuefeng's avatar
Votenet  
wuyuefeng committed
8 
]