bugfix: remove person class filter
@@ -116,7 +116,6 @@ public class DetectorFloatYoloV4 extends Detector {
}
}
final float score = maxClass;
- if (classId == 0) { // only consider persons
final float xPos = outputLocations[0][i][0];
final float yPos = outputLocations[0][i][1];
final float w = outputLocations[0][i][2];
@@ -131,7 +130,6 @@ public class DetectorFloatYoloV4 extends Detector {
recognitions.add(new Recognition("" + i, labels.get(classId), score, detection, classId));
}
}
- }
return nms(recognitions);
}
}
