【BUG解决】Yolov7训练 yolov7-e6e 出现IndexError: list index out of range

问题

  File "train.py", line 363, in train
    loss, loss_items = compute_loss_ota(pred, targets.to(device), imgs)  # loss scaled by batch_size
  File "/mnt/VISDRONE/yolov7/yolov7-main/utils/loss.py", line 585, in __call__
    loss, loss_items = compute_loss_ota(pred, targets.to(device), imgs)  # loss scaled by batch_size
  File "/mnt/VISDRONE/yolov7/yolov7-main/utils/loss.py", line 585, in __call__
    bs, as_, gjs, gis, targets, anchors = self.build_targets(p, targets, imgs)
  File "/mnt/VISDRONE/yolov7/yolov7-main/utils/loss.py", line 677, in build_targets
    bs, as_, gjs, gis, targets, anchors = self.build_targets(p, targets, imgs)
  File "/mnt/VISDRONE/yolov7/yolov7-main/utils/loss.py", line 677, in build_targets
    b, a, gj, gi = indices[i]
IndexError: list index out of range

解决方法

使用train_aux.py 来训练yolov7-e6e.