文件名称:adaptive-wing-loss:自适应机翼损耗和加权自适应机翼损耗的实现
文件大小:2KB
文件格式:ZIP
更新时间:2024-04-15 18:36:33
Python
自适应机翼损失 自适应机翼损耗和加权自适应机翼损耗的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
【文件预览】:
adaptive-wing-loss-master
----adaptiveWingLoss.py(2KB)
----README.md(1KB)