Need to edit in YOLO Classes

AI & Data Science Deep Learning (Training & Inference) TensorRT
1

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.

4

Hi,

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