Skip to content

Commit 02bdb6d

Browse files
committed
remove reference to yolo for now
1 parent 8864cb5 commit 02bdb6d

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

src/ObjectDetector/index.js

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ import handleArguments from "../utils/handleArguments";
1313
import callCallback from "../utils/callcallback";
1414
import { mediaReady } from "../utils/imageUtilities";
1515

16-
const MODEL_OPTIONS = ["cocossd", "yolo"]; // Expandable for other models like YOLO
16+
const MODEL_OPTIONS = ["cocossd"]; // Expandable for other models like YOLO
1717

1818
class ObjectDetector {
1919
/**
@@ -24,7 +24,7 @@ class ObjectDetector {
2424
*/
2525
/**
2626
* Create ObjectDetector model. Works on video and images.
27-
* @param {string} modelNameOrUrl - The name or the URL of the model to use. Current model name options are: 'YOLO' and 'CocoSsd'.
27+
* @param {string} modelNameOrUrl - The name or the URL of the model to use.
2828
* @param {Object} options - Optional. A set of options.
2929
* @param {function} callback - Optional. A callback function that is called once the model has loaded.
3030
*/
@@ -49,9 +49,6 @@ class ObjectDetector {
4949
case "cocossd":
5050
this.modelToUse = cocoSsd;
5151
break;
52-
case "yolo":
53-
this.modelToUse = yolo;
54-
break;
5552
// more models... currently only cocossd is supported
5653
default:
5754
console.warn(`Unknown model: ${this.modelName}, defaulting to CocoSsd`);

0 commit comments

Comments
 (0)