Check for accurate presence

Author: Kevin Scott

src/index.ts

@@ -2,6 +2,7 @@ import * as tf from '@tensorflow/tfjs';
 import cropAndResizeImage from './cropAndResizeImage';
 import getClasses from './getClasses';
 import train from './train';
+import translateImages from './translateImages';
 import loadPretrainedModel, {
   PRETRAINED_MODELS_KEYS,
 } from './loadPretrainedModel';
@@ -58,7 +59,7 @@ class MLClassifier {
   }
 
   private getData = async (dataType: string): Promise<ICollectedData> => {
-    if (dataType !== 'train' || dataType !== 'eval') {
+    if (dataType !== 'train' && dataType !== 'eval') {
       throw new Error(`Datatype ${dataType} unsupported`);
     }
 

src/translateImages.ts

@@ -1,3 +1,5 @@
+import * as tf from '@tensorflow/tfjs';
+
 const loadImage = async (src: string) => new Promise<HTMLImageElement>((resolve, reject) => {
   const image = new Image();
   image.src = src;
@@ -14,7 +16,7 @@ const translateImages = async (origImages: Array<tf.Tensor3D | HTMLImageElement
     if (typeof image === 'string') {
       const loadedImage = await loadImage(image);
       image = tf.fromPixels(loadedImage);
-    } else if (image.src) {
+    } else if (image instanceof HTMLImageElement) {
       const loadedImage = await loadImage(image.src);
       image = tf.fromPixels(loadedImage);
     }
@@ -26,4 +28,4 @@ const translateImages = async (origImages: Array<tf.Tensor3D | HTMLImageElement
   return images;
 }
 
-export translateImages;
+export default translateImages;