Update app.py

app.py

@@ -80,16 +80,17 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
         raw_video = cv2.VideoCapture(filename)
         frame_width, frame_height = int(raw_video.get(cv2.CAP_PROP_FRAME_WIDTH)), int(raw_video.get(cv2.CAP_PROP_FRAME_HEIGHT))
         frame_rate = int(raw_video.get(cv2.CAP_PROP_FPS))
-        print('Frame rate: ', str(frame_rate))
-        output_width = frame_width * 2 + margin_width
+        cframes = int(raw_video.get(cv2.CAP_PROP_FRAME_COUNT))
+        print(f'frames: {cframes}, fps: {frame_rate}')
+        # output_width = frame_width * 2 + margin_width
         
-        filename = os.path.basename(filename)
+        #filename = os.path.basename(filename)
         # output_path = os.path.join(outdir, filename[:filename.rfind('.')] + '_video_depth.mp4')
-        with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tmpfile:
-            output_path = tmpfile.name
+        #with tempfile.NamedTemporaryFile(delete=False, suffix='.mp4') as tmpfile:
+        #    output_path = tmpfile.name
         #out = cv2.VideoWriter(output_path, cv2.VideoWriter_fourcc(*"avc1"), frame_rate, (output_width, frame_height))
-        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-        out = cv2.VideoWriter(output_path, fourcc, frame_rate, (output_width, frame_height))
+        #fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+        #out = cv2.VideoWriter(output_path, fourcc, frame_rate, (output_width, frame_height))
         count=0
         depth_frames = []
         while raw_video.isOpened():
@@ -111,20 +112,21 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             
             depth = depth.cpu().numpy().astype(np.uint8)
             depth_color = cv2.applyColorMap(depth, cv2.COLORMAP_INFERNO)
+            #depth_color = cv2.cvtColor(depth_color, cv2.COLOR_GRAY2RGB)
             
-            split_region = np.ones((frame_height, margin_width, 3), dtype=np.uint8) * 255
-            combined_frame = cv2.hconcat([raw_frame, split_region, depth_color])
+            # split_region = np.ones((frame_height, margin_width, 3), dtype=np.uint8) * 255
+            # combined_frame = cv2.hconcat([raw_frame, split_region, depth_color])
             
             # out.write(combined_frame)
             # frame_path = os.path.join(temp_frame_dir, f"frame_{count:05d}.png")
             # cv2.imwrite(frame_path, combined_frame)
-            cv2.imwrite(f"frame{count}.jpg", combined_frame)
+            cv2.imwrite(f"frame{count}.jpg", depth_color)
             depth_frames.append(f"frame{count}.jpg")
             count += 1
 
         final_vid = create_video(depth_frames, frame_rate, "depth")
         raw_video.release()
-        out.release()
+        # out.release()
         cv2.destroyAllWindows()
         
         return final_vid #output_path