error when training yolov7-mask

[mrljwlm]

I train the yolov7-mask using the following command: python train.py --project D:\train_log --workers 2 --device 0 --batch-size 8 --data data/coco.yaml --img 1024 1024 --cfg cfg/yolov7-mask.yaml --weights '' --name yolov7_mask --hyp data/hyp.scratch.mask.yaml

the following error occurred: Starting training for 300 epochs...

 Epoch   gpu_mem       box       obj       cls     total    labels  img_size

0%| | 0/333 [00:05<?, ?it/s] Traceback (most recent call last): File "train.py", line 613, in train(hyp, opt, device, tb_writer) File "train.py", line 365, in train loss, loss_items = compute_loss(pred, targets.to(device)) # loss scaled by batch_size File "I:\yolov7-mask\utils\loss.py", line 453, in call tcls, tbox, indices, anchors = self.build_targets(p, targets) # targets File "I:\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

how can i fix this error?

[13354236170]

我也遇到了这个问题，请问您这边解决了吗

[WongKinYiu]

Training code of yolov7-mask is not yet released (due to detectron2 dependency), currently u7 branch support training of yolov7-seg.

[caisongan]

我也遇到了这个问题，请问您这边解决了吗 解决没？

[LiamHO1999]

Hi , Is training code of yolov7-mask released now?