def __init__(self, net):
super(NetworkWrapper, self).__init__()
self.net = net
self.act_crit = net_utils.FocalLoss()
self.awh_crit = net_utils.IndL1Loss1d('smooth_l1')
self.cp_crit = net_utils.FocalLoss()
self.cp_wh_crit = net_utils.IndL1Loss1d('smooth_l1')
def __init__(self, net):
super(NetworkWrapper, self).__init__()
self.net = net
self.ct_crit = net_utils.FocalLoss() #这几个是损失函数的初始化
self.wh_crit = net_utils.IndL1Loss1d('smooth_l1')
self.reg_crit = net_utils.IndL1Loss1d('smooth_l1')
self.ex_crit = torch.nn.functional.smooth_l1_loss
self.py_crit = torch.nn.functional.smooth_l1_loss
def __init__(self, net):
super(NetworkWrapper, self).__init__()
self.net = net
self.ct_crit = net_utils.FocalLoss()
self.wh_crit = net_utils.IndL1Loss1d('smooth_l1')
self.reg_crit = net_utils.IndL1Loss1d('smooth_l1')
self.ex_crit = torch.nn.functional.smooth_l1_loss
self.py_crit = torch.nn.functional.smooth_l1_loss
if cfg.poly_cls_branch:
self.py_cls_crit = torch.nn.functional.cross_entropy
if cfg.dist_constraint:
self.dist_constraint_crit = net_utils.DistanceConstraint()