Bug while trying to train instance segmentation model on custom dataset

[achbogga]

yolov7-mask/utils/loss.py", line 516, in build_targets
    gain[2:6] = torch.tensor(p[i].shape)[[3, 2, 3, 2]]  # xyxy gain
KeyError: 0

I have converted my coco format polygons to yolo format as explained here

I am using the mask branch from the repo. I tried to train and I am getting this error. Please help.

[achbogga]

@WongKinYiu Please take a look.

[WongKinYiu]

For training instance segmentation please use https://github.com/WongKinYiu/yolov7/tree/u7/seg.
You could take a look of https://github.com/RizwanMunawar/yolov7-segmentation for training custom dataset.

[achbogga]

Thanks for the update.