add bounding box mirroring (#176)

ml5js · Jul 23, 2024 · bac9e87 · bac9e87
1 parent 542e5a5
commit bac9e87
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/BodyPose/index.js b/src/BodyPose/index.js
@@ -269,6 +269,7 @@ class BodyPose {
     }
     if (this.runtimeConfig.flipHorizontal) {
       this.mirrorKeypoints(result, image.width);
+      this.mirrorBoundingBox(result);
     }
     this.addKeypoints(result);
     this.resizeBoundingBoxes(result, image.width, image.height);
@@ -333,6 +334,7 @@ class BodyPose {
       }
       if (this.runtimeConfig.flipHorizontal) {
         this.mirrorKeypoints(result, this.detectMedia.width);
+        this.mirrorBoundingBox(result);
       }
       this.addKeypoints(result);
       this.resizeBoundingBoxes(
@@ -388,6 +390,20 @@ class BodyPose {
     });
   }
 
+  /**
+   * Mirror the bounding box around x-axis.
+   * @param {Object} poses - the original detection results.
+   * @private
+   */
+  mirrorBoundingBox(poses) {
+    poses.forEach((pose) => {
+      if (!pose.box) return;
+      const tempXMin = pose.box.xMin;
+      pose.box.xMin = 1 - pose.box.xMax;
+      pose.box.xMax = 1 - tempXMin;
+    });
+  }
+
   /**
    * Resize the keypoints output of moveNet model to match the display size.
    *