Added in resampling of larger images

app.py

@@ -5,6 +5,7 @@ import numpy as np
 from PIL import Image
 from pathlib import Path
 from depth_viewer import depthviewer2html
+import cv2
 
 feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-large")
 model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large")
@@ -13,6 +14,10 @@ def process_image(image_path):
     image_path = Path(image_path)
     image = Image.open(image_path)
 
+    # if wider than 512 pixels let's resample to keep it performant on phones etc
+    if (image.size[0] > 512):
+        image = image.resize((512,int(512*image.size[1]/image.size[0])),Image.Resampling.LANCZOS)
+
     # prepare image for the model
     encoding = feature_extractor(image, return_tensors="pt")
 

depth_viewer.py

@@ -53,12 +53,6 @@ _viewer_html = '''
                 return document.getElementById("fshader").text
             }
 
-            // The url for the image is passed in to the web application
-            //var p = new URLSearchParams(window.location.search);
-            //var image_url = 'https://tujot.com/thr/marsattacks.jpg';//p.get("image");
-            //var depth_url = '/depth?image='+image_url;
-
-
             var texture = new THREE.TextureLoader().load(image_url, t => {
                 var w = t.image.width;
                 var h = t.image.height;