import torch
gamma = torch.ones_like(focal_weight).cuda()
gamma[focal_weight > 0.5] = 0.4
gamma[focal_weight < 0.5] = 2.2
focal_weight = alpha_factor * torch.pow(focal_weight, gamma)
pytorch focalloss
猜你喜欢
转载自blog.csdn.net/jacke121/article/details/105578908
今日推荐
周排行