模型改进-损失函数合集

模版

第一步在哪些地方做出修改：

228行 self.use_wiseiou=True

230行 self.wiou_loss = WiseIouLoss(ltype='MPDIoU', monotonous=False, inner_iou=True, focaler_iou=False)

238行  wiou = self.wiou_loss(pred_bboxes[fg_mask], target_bboxes[fg_mask], ret_iou=False, ratio=0.7, d=0.0, u=0.95, **{'mpdiou_hw':mpdiou_hw[fg_mask]}).unsqueeze(-1) # Wise-MPDIoU,Wise-Inner-MPDIoU,Wise-Focaler-MPDIoU

修改如图所示

第二步 train.py，不做修改但要截图

执行train.py 、get_FPS.py和val.py模型训练截图。

注：执行get_FPS.py和val.py一定要修改训练轮次并保存。