Multiclass labeling on YOLOV3-SPP-LSTM?

AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )

http://pjreddie.com/darknet/

Other

21.8k stars 7.97k forks source link

Multiclass labeling on YOLOV3-SPP-LSTM? #4029

Closed WadRex closed 5 years ago

WadRex commented 5 years ago

Hello, Is it possible to create a superclass and a subclass using LSTM version of yolov3-spp? For example superclass ("person") and subclasses ("jumping" or "standing") Thanks in advance!

AlexeyAB commented 5 years ago

@WadRex Hi,

There is multiclass rather than subclasses. In Yolo v3.

Just mark two bboxes with the same x,y,w,h and different class_id.

WadRex commented 5 years ago

@AlexeyAB

Will I get 2 predictions as output? First with "person" and second with either of two variants?

AlexeyAB commented 5 years ago

Yes. You will get two detections with the same x,y,w,h,probabiliry but with different class id

WadRex commented 5 years ago

Ok, thanks a lot