Need to edit in YOLO Classes

This is General Deep Learning Question. In YOLO v8 I want to only the Person class after that I will want to custom train my object and I have the dataset for that. For example, if I want to train pen and charger I want to give a custom dataset for these two objects and Yolo’s pre-defined class person.

At last, I want

  1. Person - pre-trained
  2. Pen - Custom
  3. Charger - Custom

How to archive this? Only one or some pre-trained classes of YOLO and others are my custom objects.


This forum talks more about updates and issues related to TensorRT. We recommend that you please reach out to the Yolo-related platform to get better help.
And following tutorial may help you.

Thank you.

1 Like