Change num_classes for finetuning

[juliuskuehn1]

Hello, I try to finetune the yolox-l weights. My dataset containes only 8 classes while the pretrained weights of yolox-l are trained on 80 classes. How can I change the classes on loaded model weights to my custom number of classes.

Thnx.

[tcotte]

Change /content/YOLOX/yolox/data/datasets/voc_classes.py file if you use a PASCAL_VOC dataset and type the 8 classes that you want in the VOC_CLASSES tuple.