3d visualization and req.txt
Browse files- Dockerfile +1 -5
- main.py +1 -0
- requirements.txt +3 -0
Dockerfile
CHANGED
@@ -22,15 +22,11 @@ RUN pip install -e .
|
|
22 |
WORKDIR /..
|
23 |
|
24 |
# RUN mim install "mmpose>=1.1.0"
|
25 |
-
|
26 |
-
RUN pip install --no-cache-dir --upgrade gradio
|
27 |
-
RUN pip install --no-cache-dir --upgrade numpy
|
28 |
-
RUN pip install --no-cache-dir --upgrade opencv-python
|
29 |
|
30 |
#https://stackoverflow.com/questions/55313610/importerror-libgl-so-1-cannot-open-shared-object-file-no-such-file-or-directo
|
31 |
RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
|
32 |
|
33 |
-
|
34 |
# Set up a new user named "user" with user ID 1000
|
35 |
RUN useradd -m -u 1000 user
|
36 |
|
|
|
22 |
WORKDIR /..
|
23 |
|
24 |
# RUN mim install "mmpose>=1.1.0"
|
25 |
+
RUN pip install -r requirements.txt
|
|
|
|
|
|
|
26 |
|
27 |
#https://stackoverflow.com/questions/55313610/importerror-libgl-so-1-cannot-open-shared-object-file-no-such-file-or-directo
|
28 |
RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
|
29 |
|
|
|
30 |
# Set up a new user named "user" with user ID 1000
|
31 |
RUN useradd -m -u 1000 user
|
32 |
|
main.py
CHANGED
@@ -44,6 +44,7 @@ def poses(photo):
|
|
44 |
out_writer = cv2.VideoWriter(output_file, fourcc, fps, size)
|
45 |
|
46 |
for result in result_generator:
|
|
|
47 |
frame = result["visualization"]
|
48 |
out_writer.write(cv2.cvtColor(frame[0], cv2.COLOR_BGR2RGB))
|
49 |
|
|
|
44 |
out_writer = cv2.VideoWriter(output_file, fourcc, fps, size)
|
45 |
|
46 |
for result in result_generator:
|
47 |
+
print("[INFO] Result: ", result)
|
48 |
frame = result["visualization"]
|
49 |
out_writer.write(cv2.cvtColor(frame[0], cv2.COLOR_BGR2RGB))
|
50 |
|
requirements.txt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
gradio
|
2 |
+
numpy
|
3 |
+
opencv-python
|