visualization

Dockerfile

@@ -25,6 +25,7 @@ WORKDIR /..
 
 RUN pip install --no-cache-dir --upgrade gradio
 RUN pip install --no-cache-dir --upgrade numpy
+RUN pip install --no-cache-dir --upgrade opencv-python
 
 #https://stackoverflow.com/questions/55313610/importerror-libgl-so-1-cannot-open-shared-object-file-no-such-file-or-directo
 RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y

main.py

@@ -5,6 +5,7 @@ print("[INFO]: Imported modules!")
 
 import gradio as gr
 import numpy as np
+import cv2
 
 inferencer = MMPoseInferencer('human')
 print("[INFO]: Downloaded models!")
@@ -15,15 +16,37 @@ def poses(photo):
                                   vis_out_dir =".",
                                   return_vis=True,
                                   thickness=2)
-    print(next(result_generator)["visualization"])
-    results = [result["visualization"] for result in result_generator]
-    print(results)
+    
+
+
+    video = cv2.VideoCapture(photo)
+
+    # Prepare to save video
+    output_file = os.path.join("output.mp4")
+
+    fourcc = cv2.VideoWriter_fourcc(*"mp4v")  # Codec for MP4 video
+    fps = video.get(cv2.CAP_PROP_FPS)
+    height = 480
+    width = 640
+    size = (width,height)
+
+    out_writer = cv2.VideoWriter(output_file, fourcc, fps, size)
+
+    for result in result_generator:
+        frame = result["visualization"]
+        out_writer.write(frame)
+
     print(os.listdir())
     print("[INFO]: Visualizing results!")
     print(os.listdir())
     print()
+    
+    out_writer.release()
+
+    cv2.destroyAllWindows() # Closing window
+
 
-    return "000000.avi"
+    return output_file
 
 
 # # specify detection model by alias