3d visualization and req.txt

Dockerfile

@@ -22,15 +22,11 @@ RUN pip install -e .
 WORKDIR /..
 
 # RUN mim install "mmpose>=1.1.0"
-
-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
+RUN pip install -r requirements.txt
 
 #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
 
-
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 

main.py

@@ -44,6 +44,7 @@ def poses(photo):
     out_writer = cv2.VideoWriter(output_file, fourcc, fps, size)
 
     for result in result_generator:
+        print("[INFO] Result: ", result)
         frame = result["visualization"]
         out_writer.write(cv2.cvtColor(frame[0], cv2.COLOR_BGR2RGB))
 

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+numpy
+opencv-python