Update app.py

app.py

@@ -172,7 +172,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
             frame_pil = Image.fromarray((frame * 255).astype(np.uint8))
             #frame = transform({'image': frame})['image']
             #frame = torch.from_numpy(frame).unsqueeze(0).to(DEVICE)
-            raw_frame_bg = cv2.medianBlur(raw_frame, 255)
+            #raw_frame_bg = cv2.medianBlur(raw_frame, 255)
 
             #
             depth = predict_depth(raw_frame[:, :, ::-1], model)
@@ -223,20 +223,20 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits', blur_data
 
             blur_frame = blur_image(raw_frame, depth_color, blur_data)
             
-            cv2.imwrite(f"f{count}.png", blur_frame)
-            orig_frames.append(f"f{count}.png")
+            cv2.imwrite(f"f{count}.jpg", blur_frame)
+            orig_frames.append(f"f{count}.jpg")
 
-            cv2.imwrite(f"f{count}_.png", blur_frame)
-            backup_frames.append(f"f{count}_.png")
+            cv2.imwrite(f"f{count}_.jpg", blur_frame)
+            backup_frames.append(f"f{count}_.jpg")
             
-            cv2.imwrite(f"f{count}_dmap.png", depth_color)
-            depth_frames.append(f"f{count}_dmap.png")
+            cv2.imwrite(f"f{count}_dmap.jpg", depth_color)
+            depth_frames.append(f"f{count}_dmap.jpg")
 
             depth_gray = seg_frame(depth_gray, o, b) + 128
             #print(depth_gray[depth_gray>128]-128)
 
-            cv2.imwrite(f"f{count}_mask.png", depth_gray)
-            masks.append(f"f{count}_mask.png")            
+            cv2.imwrite(f"f{count}_mask.jpg", depth_gray)
+            masks.append(f"f{count}_mask.jpg")            
             count += 1
 
         final_vid = create_video(orig_frames, frame_rate, "orig")
@@ -654,7 +654,7 @@ async(c, o, p, d, n, m)=>{
           if (o != "") { o += ","; }
           o += n;
         }
-        alert(o);
+        //alert(o);
         var o_ = o.split(",");
         document.getElementById("iframe3D").contentDocument.getElementById("coords").value = c;
         document.getElementById("iframe3D").contentDocument.getElementById("order").value = o;
@@ -662,7 +662,7 @@ async(c, o, p, d, n, m)=>{
         document.getElementById("iframe3D").contentDocument.getElementById("bgdepth").value = "";
         for (var j=0; j<o_.length; j++) {
           o_[j] = parseInt(o_[j]);
-          alert(o_[j]);
+          //alert(o_[j]);
           document.getElementById("iframe3D").contentDocument.getElementById("bgimage").value += p[o_[j]].image.url + ",";
           document.getElementById("iframe3D").contentDocument.getElementById("bgdepth").value += m[o_[j]].url + ",";
         }