search

ivilson / Yolov7net

Yolo Detector for .Net 8
83 stars 25 forks source link

yolov10 only detects 1 class #34

Closed Adri1996PowerGirl closed 3 weeks ago

Adri1996PowerGirl commented 4 weeks ago

all works perfect, but the output is always only one "prediction". if there is a bench and a person in an image, it detects only one of them.

Adri1996PowerGirl commented 4 weeks ago

i think the "ParseDetect" must be changed to:

            for (int i = 0; i < output.Dimensions[1]; i++)
            {
                var span = output.Buffer.Span.Slice(i * output.Strides[1]);
iwaitu commented 3 weeks ago

new release 1.0.10 fixed the bug where YOLOv10 inference results only contained one prediction. thanks for your help.

iwaitu commented 3 weeks ago

Adri1996PowerGirl commented 3 weeks ago

thank you for the fast fix. I appreciate your work!