fast-reid/fastreid/modeling/losses/__init__.py

13 lines
319 B
Python

# encoding: utf-8
"""
@author: l1aoxingyu
@contact: sherlockliao01@gmail.com
"""
from .circle_loss import *
from .cross_entroy_loss import cross_entropy_loss, log_accuracy
from .focal_loss import focal_loss
from .triplet_loss import triplet_loss
__all__ = [k for k in globals().keys() if k not in k.startswith("_")]