cuda
Browse files- Dockerfile +2 -0
- main.py +8 -8
- requirements.txt +3 -1
Dockerfile
CHANGED
@@ -6,6 +6,8 @@ COPY ./requirements.txt /code/requirements.txt
|
|
6 |
#RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.ht
|
7 |
#RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht -extra-index-url https://download.pytorch.org/whl/cu111
|
8 |
#https://download.pytorch.org/whl/torch_stable.ht
|
|
|
|
|
9 |
|
10 |
RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cu113
|
11 |
# --index-url https://download.pytorch.org/whl/cu117
|
|
|
6 |
#RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.ht
|
7 |
#RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht -extra-index-url https://download.pytorch.org/whl/cu111
|
8 |
#https://download.pytorch.org/whl/torch_stable.ht
|
9 |
+
RUN pip uninstall torch -y
|
10 |
+
RUN pip uninstall torchvision -y
|
11 |
|
12 |
RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cu113
|
13 |
# --index-url https://download.pytorch.org/whl/cu117
|
main.py
CHANGED
@@ -189,10 +189,10 @@ def run_UI():
|
|
189 |
# Insert slider with kpt_thr
|
190 |
file_kpthr = gr.Slider(minimum=0.1, maximum=1, step=20, default=0.3, label='Keypoint threshold')
|
191 |
|
192 |
-
submit_pose_file = gr.Button("Make 2d pose estimation")
|
193 |
-
submit_pose3d_file = gr.Button("Make 3d pose estimation")
|
194 |
-
submit_hand_file = gr.Button("Make 2d hand estimation")
|
195 |
-
submit_detect_file = gr.Button("Detect and track objects")
|
196 |
|
197 |
with gr.Row():
|
198 |
video_output1 = gr.PlayableVideo(height=512, label = "Estimate human 2d poses", show_label=True)
|
@@ -209,10 +209,10 @@ def run_UI():
|
|
209 |
|
210 |
web_kpthr = gr.Slider(minimum=0.1, maximum=1, step=20, default=0.3, label='Keypoint threshold')
|
211 |
|
212 |
-
submit_pose_web = gr.Button("Make 2d pose estimation")
|
213 |
-
submit_pose3d_web = gr.Button("Make 3d pose estimation")
|
214 |
-
submit_hand_web = gr.Button("Make 2d hand estimation")
|
215 |
-
submit_detect_web = gr.Button("Detect and track objects")
|
216 |
with gr.Row():
|
217 |
webcam_output1 = gr.PlayableVideo(height=512, label = "Estimate human 2d poses", show_label=True)
|
218 |
webcam_output2 = gr.PlayableVideo(height=512, label = "Estimate human 3d poses", show_label=True)
|
|
|
189 |
# Insert slider with kpt_thr
|
190 |
file_kpthr = gr.Slider(minimum=0.1, maximum=1, step=20, default=0.3, label='Keypoint threshold')
|
191 |
|
192 |
+
submit_pose_file = gr.Button("Make 2d pose estimation", variant="primary")
|
193 |
+
submit_pose3d_file = gr.Button("Make 3d pose estimation", variant="primary")
|
194 |
+
submit_hand_file = gr.Button("Make 2d hand estimation", variant="primary")
|
195 |
+
submit_detect_file = gr.Button("Detect and track objects", variant="primary")
|
196 |
|
197 |
with gr.Row():
|
198 |
video_output1 = gr.PlayableVideo(height=512, label = "Estimate human 2d poses", show_label=True)
|
|
|
209 |
|
210 |
web_kpthr = gr.Slider(minimum=0.1, maximum=1, step=20, default=0.3, label='Keypoint threshold')
|
211 |
|
212 |
+
submit_pose_web = gr.Button("Make 2d pose estimation", variant="primary")
|
213 |
+
submit_pose3d_web = gr.Button("Make 3d pose estimation", variant="primary")
|
214 |
+
submit_hand_web = gr.Button("Make 2d hand estimation", variant="primary")
|
215 |
+
submit_detect_web = gr.Button("Detect and track objects", variant="primary")
|
216 |
with gr.Row():
|
217 |
webcam_output1 = gr.PlayableVideo(height=512, label = "Estimate human 2d poses", show_label=True)
|
218 |
webcam_output2 = gr.PlayableVideo(height=512, label = "Estimate human 3d poses", show_label=True)
|
requirements.txt
CHANGED
@@ -3,4 +3,6 @@ numpy
|
|
3 |
opencv-python
|
4 |
ultralytics
|
5 |
lap
|
6 |
-
MoviePy
|
|
|
|
|
|
3 |
opencv-python
|
4 |
ultralytics
|
5 |
lap
|
6 |
+
MoviePy
|
7 |
+
torch
|
8 |
+
torchvision
|