Rename last remnants of BodyPix

This fixes the body segmentation examples in #81.
ml5js · Mar 16, 2024 · 51a5d72 · 51a5d72
1 parent 13ba73c
commit 51a5d72
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/documentation.md b/documentation.md
@@ -22,7 +22,7 @@ const bodySegmentation = ml5.bodySegmentation(?modelName, ?options, ?callback);
 
 **Parameters:**
 
-- **modelName**: OPTIONAL: A string specifying which model to use, "SelfieSegmentation" or "BodyPix".
+- **modelName**: OPTIONAL: A string specifying which model to use, "SelfieSegmentation" or "BodySegmentation".
 
 - **options**: OPTIONAL. An object to change the default configuration of the model. See the example options:
 

diff --git a/src/index.js b/src/index.js
@@ -7,7 +7,7 @@ import * as tf from "@tensorflow/tfjs";
 import * as tfvis from "@tensorflow/tfjs-vis";
 import p5Utils from "./utils/p5Utils";
 import setBackend from "./utils/setBackend";
-import bodyPix from "./BodySegmentation";
+import bodySegmentation from "./BodySegmentation";
 import communityStatement from "./utils/communityStatement";
 import imageClassifier from "./ImageClassifier";
 import preloadRegister from "./utils/p5PreloadHelper";
@@ -25,7 +25,7 @@ export default Object.assign({ p5Utils }, preloadRegister(withPreload), {
   faceMesh,
   bodyPose,
   setBackend,
-  bodyPix,
+  bodySegmentation,
 });
 
 communityStatement();