search

AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.75k stars 7.96k forks source link

Training Sub Categories #4546

Open nedkelly734 opened 4 years ago

nedkelly734 commented 4 years ago

Just wondering is it possible to train sub classes of object categories with yolov3 or yolov2 as I want to train my detection system to recognise various model of cars, planes, and trains (i.e. Car as a parent class and models of cars)? During my research I notice that the previous yolo9000 does similarly to want I want and I also recognised that certain databases such as openimages are set up in a similarly manner but I am unable to find a tutorial or instructions on how to setup sub classes to train with? Else if its not possible to have subclasses in yolov2 or yolov3, how can one train yolo9000 as there seems only to be information on training specific objects and not sub categories of objects. Any help would be appreciated

Patrick-PhoenixAI commented 4 years ago

I found the following post useful: How to implement multi-label detection?