How can I train only the person class?

WongKinYiu / yolov9

Implementation of paper - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information

GNU General Public License v3.0

8.95k stars 1.42k forks source link

How can I train only the person class? #286

Open alsrlwjs56 opened 7 months ago

alsrlwjs56 commented 7 months ago

I am using cocodataset, I don't want to train any other classes besides humans.

How do I train only person classes with yolov9? yolov9-c.yaml > Changed nc: 80 to nc: 1. coco.yaml > names: Comment all except person args --single-cls Even if you change the above, train_batch still recognizes all classes as 0.

TheMattBin commented 6 months ago

maybe you can refer to #30 and check if it help