How to solve this problem

[huanai]

[a28293971]

check your anchors size and class num if you tain your custom dataset.
please note that, there are many const value in the class Yolov4 and function train, you also need to change them to the correct value which fit for your dataset.
e.g. anchors, image_size or n_classes...

[Mulbetty]

Hi, how do you solve the problem? Thank you

[a28293971]

Hi, how do you solve the problem? Thank you

As I said above, you need to understand the theory of Yolov4 first, and then go to the source code to modify the parts that need to be fitted for your dataset size.

[ErlingLie]

You should try this solution #138 .

You need to change the classes values in the cfg so that it matches your number of classes, and also the number of filters in the previous layer such that it equals 3*(5 + num classes) instead of 255.