AttributeError: 'list' object has no attribute 'shape'

[Ureisen]

Hi!
I encountered such an error in the training process of using my own dataset (exported by RoboFlow). Is it because I have only one segmentation object?
It will be appreciated if you could give me some suggestions！

Traceback (most recent call last):
File "train.py", line 613, in
main(opt)
File "train.py", line 509, in main
train(opt.hyp, opt, device, callbacks)
File "train.py", line 291, in train
loss, loss_items = compute_loss(pred, targets.to(device)) # loss scaled by batch_size
File "/home/yolov7-segmentation-main/utils/loss.py", line 127, in call
tcls, tbox, indices, anchors = self.build_targets(p, targets) # targets
File "/home/yolov7-segmentation-main/utils/loss.py", line 200, in build_targets
anchors, shape = self.anchors[i], p[i].shape
AttributeError: 'list' object has no attribute 'shape'

[AndrewTKent]

@Ureisen I have the same issue, did you figure out the solution?

[RizwanMunawar]

@AndrewTKent! What is the format of your labels, its in coco format or YOLO format?

[AndrewTKent]

@RizwanMunawar Hi! Yes I have it in coco format (JSON) and I'm working out of a folder called coco_custom, and I have a coco_custom.yaml pointing to this folder

coco_custom
│--annotations
| |--instances_default.json
│--images
| |--train
| | |--.jpg
| |--val
| | |--.jpg
│--train.txt
│--val.txt

[goeunnykim]

@AndrewTKent! What is the format of your labels, its in coco format or YOLO format?

Hi! Is it fixed? I have the same issue. My dataset is in yolo format.

[RizwanMunawar]

@RizwanMunawar Hi! Yes I have it in coco format (JSON) and I'm working out of a folder called coco_custom, and I have a coco_custom.yaml pointing to this folder

coco_custom │--annotations | |--instances_default.json │--images | |--train | | |--.jpg | |--val | | |--.jpg │--train.txt │--val.txt

@AndrewTKent The format is wrong, you will need to convert the labels to yolo (Polygon box) format. Also I will recommend to follow the tutorial for data labeling, that I already recommended for proper training.
https://medium.com/augmented-startups/train-yolov7-segmentation-on-custom-data-b91237bd2a29

[RizwanMunawar]

@AndrewTKent! What is the format of your labels, its in coco format or YOLO format?

Hi! Is it fixed? I have the same issue. My dataset is in yolo format.

The required labeling format is Polygon box (with YOLO format). You can follow the tutorial for data labeling, that I already recommended for proper training.
https://medium.com/augmented-startups/train-yolov7-segmentation-on-custom-data-b91237bd2a29

[stephansturges]

I have this with the existing code base as well, with labels in YOLO format... digging into it to propose a fix.

[5JM]

It's late, but I'll leave it here in case anyone gets lost.
If an error occurs even in YOLO format

Check you are running ./segment/train.py.
(Not ./train.py)