Closed devloper13 closed 3 years ago
你好,我也遇到这个问题,请问您找到解决方法了吗
Hey, how did you solve this?
How did you solve this? I have the same problem, too.
In coco.names, there is an empty line at the last.
maybe you didn't create the last empty line. It worked for me after added this line.
I have a dataset of my own which has 8 labels, completely different from the coco labels. I changed the data/coco.names and data/coco.yaml accordingly. But I get an index error:
I tried printing the detection classes and its in the range 0-79 i.e coco labels. But why is this happening when I completely changed the labels?
Training command:
python train.py --batch-size 1 --img 1280 1280 --data coco.yaml --cfg cfg/yolor_p6.cfg --weights yolor_p6.pt --device 0 --name yolor_p6_digit --hyp hyp.scratch.1280.yaml --epochs 5