@@ -78,13 +78,15 @@ class FaceApiBase {
78
78
} = this . config . MODEL_URLS ;
79
79
80
80
this . model = faceapi ;
81
-
82
- const SsdMobilenetv1Options = this . model . SsdMobilenetv1Options ( { minConfidence : this . minConfidence } )
81
+
82
+ const SsdMobilenetv1Options = this . model . SsdMobilenetv1Options ( {
83
+ minConfidence : this . minConfidence
84
+ } )
83
85
await this . model . loadSsdMobilenetv1Model ( Mobilenetv1Model , SsdMobilenetv1Options )
84
86
await this . model . loadFaceLandmarkModel ( FaceLandmarkModel )
85
87
// await this.model.loadFaceLandmarkTinyModel(FaceLandmark68TinyNet)
86
88
await this . model . loadFaceRecognitionModel ( FaceRecognitionModel )
87
-
89
+
88
90
this . modelReady = true ;
89
91
return this ;
90
92
}
@@ -101,18 +103,20 @@ class FaceApiBase {
101
103
let callback ;
102
104
let faceApiOptions = this . config ;
103
105
104
- // Handle the image to predict
105
- if ( typeof optionsOrCallback === 'function' ) {
106
+ // Handle the image to predict
107
+ if ( typeof optionsOrCallback === 'function' ) {
106
108
imgToClassify = this . video ;
107
109
callback = optionsOrCallback ;
108
110
// clean the following conditional statement up!
109
- } else if ( optionsOrCallback instanceof HTMLImageElement
110
- || optionsOrCallback instanceof HTMLCanvasElement
111
- || optionsOrCallback instanceof ImageData ) {
112
- imgToClassify = optionsOrCallback ;
113
- } else if ( typeof optionsOrCallback === 'object' && ( optionsOrCallback . elt instanceof HTMLImageElement
114
- || optionsOrCallback . elt instanceof HTMLCanvasElement
115
- || optionsOrCallback . elt instanceof ImageData ) ) {
111
+ } else if ( optionsOrCallback instanceof HTMLImageElement ||
112
+ optionsOrCallback instanceof HTMLCanvasElement ||
113
+ optionsOrCallback instanceof HTMLVideoElement ||
114
+ optionsOrCallback instanceof ImageData ) {
115
+ imgToClassify = optionsOrCallback ;
116
+ } else if ( typeof optionsOrCallback === 'object' && ( optionsOrCallback . elt instanceof HTMLImageElement ||
117
+ optionsOrCallback . elt instanceof HTMLCanvasElement ||
118
+ optionsOrCallback . elt instanceof HTMLVideoElement ||
119
+ optionsOrCallback . elt instanceof ImageData ) ) {
116
120
imgToClassify = optionsOrCallback . elt ; // Handle p5.js image
117
121
} else if ( typeof optionsOrCallback === 'object' && optionsOrCallback . canvas instanceof HTMLCanvasElement ) {
118
122
imgToClassify = optionsOrCallback . canvas ; // Handle p5.js image
@@ -199,13 +203,15 @@ class FaceApiBase {
199
203
imgToClassify = this . video ;
200
204
callback = optionsOrCallback ;
201
205
// clean the following conditional statement up!
202
- } else if ( optionsOrCallback instanceof HTMLImageElement
203
- || optionsOrCallback instanceof HTMLCanvasElement
204
- || optionsOrCallback instanceof ImageData ) {
205
- imgToClassify = optionsOrCallback ;
206
- } else if ( typeof optionsOrCallback === 'object' && ( optionsOrCallback . elt instanceof HTMLImageElement
207
- || optionsOrCallback . elt instanceof HTMLCanvasElement
208
- || optionsOrCallback . elt instanceof ImageData ) ) {
206
+ } else if ( optionsOrCallback instanceof HTMLImageElement ||
207
+ optionsOrCallback instanceof HTMLCanvasElement ||
208
+ optionsOrCallback instanceof HTMLVideoElement ||
209
+ optionsOrCallback instanceof ImageData ) {
210
+ imgToClassify = optionsOrCallback ;
211
+ } else if ( typeof optionsOrCallback === 'object' && ( optionsOrCallback . elt instanceof HTMLImageElement ||
212
+ optionsOrCallback . elt instanceof HTMLCanvasElement ||
213
+ optionsOrCallback . elt instanceof HTMLVideoElement ||
214
+ optionsOrCallback . elt instanceof ImageData ) ) {
209
215
imgToClassify = optionsOrCallback . elt ; // Handle p5.js image
210
216
} else if ( typeof optionsOrCallback === 'object' && optionsOrCallback . canvas instanceof HTMLCanvasElement ) {
211
217
imgToClassify = optionsOrCallback . canvas ; // Handle p5.js image
0 commit comments