模版
第一步在哪些地方做出修改:
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一定要修改训练轮次并保存。