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()