尝试使用 colab notebooks 训练 yolov5 和 v8 后出现 yolo 分段错误

Question

**代码-!python /content/yolov5/segment/train.py --img 320 --batch 128 --epochs 100 --data /content/drive/MyDrive/lungcancer_project/dataset.yaml --weights yolov5m-seg .pt

文件“/content/yolov5/utils/segment/augmentations.py”，第 102 行，在 random_perspective new_segments = np.array(new_segments)[i]

IndexError：布尔索引与维度 0 上的索引数组不匹配；维度为 0 但相应的布尔维度为 8 **