patch style transfer for non-square and canvas-derived inputs (#311)

Author: JazzTap

src/StyleTransfer/index.js

@@ -89,7 +89,9 @@ class StyleTransfer extends Video {
     let input;
     let callback = cb;
 
-    if (inputOrCallback instanceof HTMLVideoElement || inputOrCallback instanceof HTMLImageElement) {
+    if (inputOrCallback instanceof HTMLVideoElement ||
+        inputOrCallback instanceof HTMLImageElement ||
+        inputOrCallback instanceof ImageData) {
       input = inputOrCallback;
     } else if (typeof inputOrCallback === 'object' && (inputOrCallback.elt instanceof HTMLVideoElement || inputOrCallback.elt instanceof HTMLImageElement)) {
       input = inputOrCallback.elt;

src/utils/imageUtilities.js

@@ -24,7 +24,7 @@ const processVideo = (input, size, callback = () => {}) => {
 
 // Converts a tf to DOM img
 const array3DToImage = (tensor) => {
-  const [imgWidth, imgHeight] = tensor.shape;
+  const [imgHeight, imgWidth] = tensor.shape;
   const data = tensor.dataSync();
   const canvas = document.createElement('canvas');
   canvas.width = imgWidth;