fix - adds callback support for single functions in posenet re: #244 (#254)

* fix(adds callback support for single functions in posenet https://github.com/ml5js/ml5-library/issue

adds callback support for single functions in posenet
#244

#244

* rm package-lock.json

* fix(PoseNet/index.js): Add support for PoseNet `.singlePose()` & `.multiPose()` callback #244

Add support for PoseNet `.singlePose()` & `.multiPose()` callback #244 & updated callback error
handling

#244

* rm error check when returning callback in posenet singlePose & multiPose

* fixed indentation

* temp rm cb function

* added cb function back in

* added yarn and package lock

* added lock files from ml5 master

Author: joeyklee

src/PoseNet/index.js

@@ -62,7 +62,7 @@ class PoseNet extends EventEmitter {
   }
 
   /* eslint max-len: ["error", { "code": 180 }] */
-  async singlePose(inputOr) {
+  async singlePose(inputOr, cb) {
     let input;
 
     if (inputOr instanceof HTMLImageElement || inputOr instanceof HTMLVideoElement) {
@@ -74,15 +74,22 @@ class PoseNet extends EventEmitter {
     }
 
     const pose = await this.net.estimateSinglePose(input, this.imageScaleFactor, this.flipHorizontal, this.outputStride);
+
     const result = [{ pose, skeleton: this.skeleton(pose.keypoints) }];
     this.emit('pose', result);
+
     if (this.video) {
       return tf.nextFrame().then(() => this.singlePose());
     }
+
+    if (typeof cb === 'function') {
+      cb(result);
+    }
+
     return result;
   }
 
-  async multiPose(inputOr) {
+  async multiPose(inputOr, cb) {
     let input;
 
     if (inputOr instanceof HTMLImageElement || inputOr instanceof HTMLVideoElement) {
@@ -99,6 +106,11 @@ class PoseNet extends EventEmitter {
     if (this.video) {
       return tf.nextFrame().then(() => this.multiPose());
     }
+
+    if (typeof cb === 'function') {
+      cb(result);
+    }
+
     return result;
   }
 }

src/index.js

@@ -32,3 +32,4 @@ module.exports = {
   ...imageUtils,
   tf,
 };
+