Pytorch-yolov3: About a line in train.py: ap_table += [[c, class_names[c], "%.5f" % AP[i]]]

[Adoreee]

I use the DOTA dataset (15 classes)
when i run train.py, appear

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

Why is that?
Thank U

[Adoreee]

solved. Because the last line of the names file is missing a newline.
Thank you

[sdustdk1427]

@Adoreee Hello, I have the same problem but I don't know what you mean?How exactly did you change it?thanks！

[Adoreee]

@Adoreee Hello, I have the same problem but I don't know what you mean?How exactly did you change it?thanks！

Add a "\n" to the last line of *.names

[sdustdk1427]

@Adoreee Thank you very much!

[bisonliao]

add check code can solve?

            if c < len(class_names):
                ap_table += [[c, class_names[c], "%.5f" % AP[i]]]
            else:
                print("!!! invalid class index:", c)