fixed error in yolo example

cansik · Mar 31, 2020 · 0713df2 · 0713df2
1 parent 98b901d
commit 0713df2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/YoloWecamExample/YoloWecamExample.pde b/examples/YoloWecamExample/YoloWecamExample.pde
@@ -48,7 +48,7 @@ public void draw() {
   strokeWeight(2f);
 
   for (ObjectDetectionResult detection : detections) {
-    stroke((int)(360.0 / yolo.getClassNames().size() * detection.getClassId()), 80, 100);
+    stroke((int)(360.0 / yolo.getLabels().size() * detection.getClassId()), 80, 100);
     rect(detection.getX(), detection.getY(), detection.getWidth(), detection.getHeight());
   }