Using u7-seg branch for segmentation, using a mini COCO dataset (first 25k images).
Getting an error:
Traceback (most recent call last): File "seg/train.py", line 630, in <module> main(opt) File "seg/train.py", line 526, in main train(opt.hyp, opt, device, callbacks) File "seg/train.py", line 308, in train loss, loss_items = compute_loss(pred, targets.to(device)) # loss scaled by batch_size File "/home/jupyter/yolov7/seg/utils/loss.py", line 125, in __call__ tcls, tbox, indices, anchors = self.build_targets(p, targets) # targets File "/home/jupyter/yolov7/seg/utils/loss.py", line 198, in build_targets anchors, shape = self.anchors[i], p[i].shape AttributeError: 'list' object has no attribute 'shape'
Using u7-seg branch for segmentation, using a mini COCO dataset (first 25k images). Getting an error:
Traceback (most recent call last): File "seg/train.py", line 630, in <module> main(opt) File "seg/train.py", line 526, in main train(opt.hyp, opt, device, callbacks) File "seg/train.py", line 308, in train loss, loss_items = compute_loss(pred, targets.to(device)) # loss scaled by batch_size File "/home/jupyter/yolov7/seg/utils/loss.py", line 125, in __call__ tcls, tbox, indices, anchors = self.build_targets(p, targets) # targets File "/home/jupyter/yolov7/seg/utils/loss.py", line 198, in build_targets anchors, shape = self.anchors[i], p[i].shape AttributeError: 'list' object has no attribute 'shape'
Full console output: