adaptive wing loss:自适应机翼损耗和加权自适应机翼损耗的实现 源码

自适应机翼损失 自适应机翼损耗和加权自适应机翼损耗的Pytorch实现 自适应机翼损失的性能 在我的实验中,自适应机翼损失的性能非常好。 在面部标志检测任务中,在300W有效集(全套)上,我得到了3.16%的眼距归一化平均误差(ION NME)。 用法: from adaptiveWingLoss import AdaptiveWingLoss # Instantiate the AdaptiveWingLoss like the way that instantiate MSELoss criterion = AdaptiveWingLoss(whetherWeighted=True) # criterion = torch.nn.MSELoss(size_average=True) for epoch in range(maxEpochNum): for img, heatMa