Pytorch-yolov3: IndexError: list index out of range(while training on custom dataset with two classes)

[impravin22]

Traceback (most recent call last):
File "train.py", line 173, in
ap_table += [[c, class_names[c], "%.5f" % AP[i]]]
IndexError: list index out of range

[nodiz]

Hello, in utils.py, load_classes change

    names = fp.read().split("\n")[:-1]

To

    names = fp.read().split("\n")

I don't really understand why in the loader they are not taking the last class into account, but I just changed there. Then it should work

[impravin22]

Hello, in utils.py, load_classes change

    names = fp.read().split("\n")[:-1]

To

    names = fp.read().split("\n")

I don't really understand why in the loader they are not taking the last class into account, but I just changed there. Then it should work

Thank you. It worked for me as I just press enter at the end of the line in the classes.names file. If you know how to detect the video, please do let me know.

Thanks

[swjtulinxi]

hello，why is my cls_acc==100%,

[cotyyang]

thanks,i also solved it