Transfer Learning for Custom Object Detection

[rut00]

Greetings,

I'm attempting transfer learning on the YOLOv7 model to include custom objects alongside the existing 80 classes. Despite retraining the pre-trained model, the original classes reset post-training. I have also tried freezing the layers, and modifying config files, and hyperparameter files but I am not getting hold of pre-trained classes.

Please provide insight for the same.

[dsbyprateekg]

The way you are following does not work with YOLO. You must add your new class data to the existing 80 classes dataset and train the model if you want to detect the newly added custom class with the existing 80 classes.