adjusted examples to version 0.5.0

cansik · Jan 26, 2021 · c3091c3 · c3091c3
1 parent f7793fc
commit c3091c3
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 9 deletions.
diff --git a/examples/FaceDetectorCNNWebcam/FaceDetectorCNNWebcam.pde b/examples/FaceDetectorCNNWebcam/FaceDetectorCNNWebcam.pde
@@ -44,5 +44,5 @@ public void draw() {
     rect(detection.getX(), detection.getY(), detection.getWidth(), detection.getHeight());
   }
 
-  surface.setTitle("Face Recognition Test - FPS: " + Math.round(frameRate));
+  surface.setTitle("Face Detector Test - FPS: " + Math.round(frameRate));
 }
diff --git a/examples/RealSensePoseCapture/RealSensePoseCapture.pde b/examples/RealSensePoseCapture/RealSensePoseCapture.pde
@@ -1,7 +1,6 @@
 import ch.bildspur.vision.DeepVision;
 import ch.bildspur.vision.SingleHumanPoseNetwork;
-import ch.bildspur.vision.result.HumanPoseResult;
-import ch.bildspur.vision.result.KeyPointResult;
+import ch.bildspur.vision.result.*;
 
 import ch.bildspur.realsense.*;
 

diff --git a/examples/RealSenseYoloDetector/RealSenseYoloDetector.pde b/examples/RealSenseYoloDetector/RealSenseYoloDetector.pde
@@ -1,6 +1,6 @@
 import ch.bildspur.vision.DeepVision;
 import ch.bildspur.vision.YOLONetwork;
-import ch.bildspur.vision.result.ObjectDetectionResult;
+import ch.bildspur.vision.result.*;
 
 import ch.bildspur.realsense.*;
 
@@ -16,7 +16,7 @@ void setup()
   colorMode(HSB, 360, 100, 100);
 
   println("creating network...");
-  net = vision.createYOLOv3Tiny();
+  net = vision.createYOLOv4Tiny();
 
   println("loading model...");
   net.setup();
@@ -51,7 +51,7 @@ void draw()
   strokeWeight(2f);
 
   for (ObjectDetectionResult detection : result) {
-    stroke(round(360.0f * (float) detection.getClassId() / net.getClassNames().size()), 75, 100);
+    stroke(round(360.0f * (float) detection.getClassId() / net.getLabels().size()), 75, 100);
     rect(detection.getX(), detection.getY(), detection.getWidth(), detection.getHeight());
 
     textSize(15);

diff --git a/examples/SingleHumanPoseEstimation/SingleHumanPoseEstimation.pde b/examples/SingleHumanPoseEstimation/SingleHumanPoseEstimation.pde
@@ -1,7 +1,6 @@
 import ch.bildspur.vision.DeepVision;
 import ch.bildspur.vision.SingleHumanPoseNetwork;
-import ch.bildspur.vision.result.HumanPoseResult;
-import ch.bildspur.vision.result.KeyPointResult;
+import ch.bildspur.vision.result.*;
 
 PImage testImage;
 

diff --git a/examples/YOLODetectObjects/YOLODetectObjects.pde b/examples/YOLODetectObjects/YOLODetectObjects.pde
@@ -15,7 +15,7 @@ public void setup() {
   image = loadImage("hk.jpg");
 
   println("creating model...");
-  yolo = deepVision.createYOLOv3();
+  yolo = deepVision.createYOLOv4();
 
   println("loading yolo model...");
   yolo.setup();