Merge branch 'main' into neuroevolution

Author: shiffman

src/index.js

@@ -4,6 +4,7 @@ import poseDetection from "./PoseDetection";
 import * as tf from "@tensorflow/tfjs";
 import * as tfvis from "@tensorflow/tfjs-vis";
 import p5Utils from "./utils/p5Utils";
+import setBackend from "./utils/setBackend";
 
 export default Object.assign(
   { p5Utils },
@@ -13,5 +14,6 @@ export default Object.assign(
     neuralNetwork,
     handpose,
     poseDetection,
+    setBackend,
   }
 );

src/utils/setBackend.js

@@ -0,0 +1,32 @@
+import * as tf from "@tensorflow/tfjs";
+
+/**
+ * Manually set the Tensorflow.js Backend to the specified backend.
+ * Determines how the operation is done for the following features:
+ * - handpose under "tfjs" runtime
+ * - poseDetection
+ * - facemesh
+ * - neuralNetwork
+ *
+ * @param {string} backend the backend to set to. Must be "cpu", "webgl", or "webgpu".
+ * @return {boolean} true if the backend was set successfully.
+ */
+export default function setBackend(backend) {
+  switch (backend) {
+    case "cpu":
+      tf.setBackend("cpu");
+      break;
+    case "webgl":
+      tf.setBackend("webgl");
+      break;
+    case "webgpu":
+      tf.setBackend("webgpu");
+      break;
+    default:
+      throw new Error(
+        `Invalid backend: ${backend}. The backend parameter must be set to one of the following strings: "cpu", "webgl", or "webgpu".`
+      );
+  }
+
+  return true;
+}