ImageClassifier: predict: handle unsupported input

Alessandro Astone · YoDevil · commit 535deb2f846b · 2018-08-08T15:40:00.000+02:00
diff --git a/src/ImageClassifier/index.js b/src/ImageClassifier/index.js
@@ -70,6 +70,9 @@ class ImageClassifier {
       imgToPredict = inputNumOrCallback;
     } else if (typeof inputNumOrCallback === 'object' && inputNumOrCallback.elt instanceof HTMLImageElement) {
       imgToPredict = inputNumOrCallback.elt; // Handle p5.js image
+    } else if (!(this.video instanceof HTMLVideoElement)) {
+      // Handle unsupported input
+      throw new Error('No input image provided. If you want to classify a video, pass the video element in the constructor. ');
     }
 
     if (typeof numOrCallback === 'number') {

Original file line number	Diff line number	Diff line change
`@@ -70,6 +70,9 @@ class ImageClassifier {`
`70`	`70`	`imgToPredict = inputNumOrCallback;`
`71`	`71`	`} else if (typeof inputNumOrCallback === 'object' && inputNumOrCallback.elt instanceof HTMLImageElement) {`
`72`	`72`	`imgToPredict = inputNumOrCallback.elt; // Handle p5.js image`
	`73`	`+ } else if (!(this.video instanceof HTMLVideoElement)) {`
	`74`	`+ // Handle unsupported input`
	`75`	`+ throw new Error('No input image provided. If you want to classify a video, pass the video element in the constructor. ');`
`73`	`76`	`}`
`74`	`77`
`75`	`78`	`if (typeof numOrCallback === 'number') {`