smoothl1函数_Python torch.nn.functional 模块，smooth_l1_loss() 实例源码 - 编程字典

def forward(self, loc_preds, loc_targets, conf_preds, conf_targets):

'''Compute loss between (loc_preds, loc_targets) and (conf_preds, conf_targets).

Args:

loc_preds: (tensor) predicted locations, sized [batch_size, 8732, 4].

loc_targets: (tensor) encoded target locations, sized [batch_size, 8732, 4].

conf_preds: (tensor) predicted class confidences, sized [batch_size, 8732, num_classes].

conf_targets: (tensor) encoded target classes, sized [batch_size, 8732].

loss:

(tensor) loss = SmoothL1Loss(loc_preds, loc_targets) + CrossEntropyLoss(conf_preds, conf_targets).

'''

batch_size, num_boxes, _ = loc_preds.size()

pos = conf_targets>0 # [N,8732], pos means the box matched.

# print(pos.size())

num_matched_boxes = pos.data.long().sum()

if num_matched_boxes == 0:

return V