adds fix for unet image return for p5 (#625)

Author: joeyklee

src/UNET/index.js

@@ -11,6 +11,7 @@ import * as tf from '@tensorflow/tfjs';
 import callCallback from '../utils/callcallback';
 import { array3DToImage } from '../utils/imageUtilities';
 import Video from '../utils/Video';
+import p5Utils from '../utils/p5Utils';
 
 const URL = 'https://raw.githubusercontent.com/zaidalyafeai/HostedModels/master/unet-128/model.json';
 const imageSize = 128;
@@ -40,10 +41,10 @@ class UNET extends Video {
   }
 
   // check if p5js
-  static checkP5() {
-    if (typeof window !== 'undefined' && window.p5 && window.p5.Image && typeof window.p5.Image === 'function') return true;
-    return false;
-  }
+  // static checkP5() {
+  //   if (typeof window !== 'undefined' && window.p5 && window.p5.Image && typeof window.p5.Image === 'function') return true;
+  //   return false;
+  // }
 
   async segment(inputOrCallback, cb) {
     await this.ready;
@@ -108,8 +109,10 @@ class UNET extends Video {
     const raw = await tf.browser.toPixels(tensor);
     let image;
 
-    if (UNET.checkP5()) {
-      image = window.loadImage(dom.src);
+    if (p5Utils.checkP5()) {
+        const blob1 = await p5Utils.rawToBlob(raw, imageSize, imageSize);
+        const p5Image1 = await p5Utils.blobToP5Image(blob1);
+        image = p5Image1;
     }
 
     return {