![]() ![]() Pred = model(img, augment=opt.augment)įile "C:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 1051, in _call_implįile "C:\Users\ASUS\Desktop\yoloR\yolor\models\models.py", line 543, in forwardįile "C:\Users\ASUS\Desktop\yoloR\yolor\models\models.py", line 594, in forward_once Hi, when I change -img-size from 1280 to 320 for example it is not working and throws me an error which is : 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 -epochs 5 But why is this happening when I completely changed the labels? I tried printing the detection classes and its in the range 0-79 i.e coco labels. Plot_images(img, output_to_target(output, width, height), paths, f, names) # predictionsįile "/home/ubuntu/yolor/utils/plots.py", line 164, in plot_images ![]() Train(hyp, opt, device, tb_writer, wandb)įile "/home/ubuntu/yolor/test.py", line 226, in test But I get an index error: Traceback (most recent call last): I changed the data/coco.names and data/coco.yaml accordingly. I have a dataset of my own which has 8 labels, completely different from the coco labels.
0 Comments
Leave a Reply. |