3d change ND 2D NEW MODEL
Browse files- Dockerfile +1 -2
- main_noweb.py +3 -1
Dockerfile
CHANGED
@@ -17,8 +17,7 @@ RUN pip3 install torchvision --extra-index-url https://download.pytorch.org/whl
|
|
17 |
RUN pip3 install --no-cache-dir --upgrade -U openmim
|
18 |
RUN mim install --no-cache-dir --upgrade mmengine
|
19 |
RUN mim install "mmcv>=2.0.1"
|
20 |
-
RUN mim install "mmdet
|
21 |
-
#RUN mim install "mmdet>=3.1.0"
|
22 |
|
23 |
RUN git clone https://github.com/open-mmlab/mmpose.git
|
24 |
|
|
|
17 |
RUN pip3 install --no-cache-dir --upgrade -U openmim
|
18 |
RUN mim install --no-cache-dir --upgrade mmengine
|
19 |
RUN mim install "mmcv>=2.0.1"
|
20 |
+
RUN mim install "mmdet>=3.1.0"
|
|
|
21 |
|
22 |
RUN git clone https://github.com/open-mmlab/mmpose.git
|
23 |
|
main_noweb.py
CHANGED
@@ -33,7 +33,9 @@ os.system("nvidia-smi")
|
|
33 |
print("[INFO]: Imported modules!")
|
34 |
human = MMPoseInferencer("simcc_mobilenetv2_wo-deconv-8xb64-210e_coco-256x192") # simcc_mobilenetv2_wo-deconv-8xb64-210e_coco-256x192 dekr_hrnet-w32_8xb10-140e_coco-512x512
|
35 |
hand = MMPoseInferencer("hand")
|
36 |
-
human3d = MMPoseInferencer("human3d")
|
|
|
|
|
37 |
#"https://download.openmmlab.com/mmpose/body3d/simple_baseline/simple3Dbaseline_h36m-f0ad73a4_20210419.pth") # pose3d="human3d"
|
38 |
#https://github.com/open-mmlab/mmpose/tree/main/configs/hand_2d_keypoint/topdown_regression
|
39 |
print("[INFO]: Downloaded models!")
|
|
|
33 |
print("[INFO]: Imported modules!")
|
34 |
human = MMPoseInferencer("simcc_mobilenetv2_wo-deconv-8xb64-210e_coco-256x192") # simcc_mobilenetv2_wo-deconv-8xb64-210e_coco-256x192 dekr_hrnet-w32_8xb10-140e_coco-512x512
|
35 |
hand = MMPoseInferencer("hand")
|
36 |
+
human3d = MMPoseInferencer("human3d")
|
37 |
+
|
38 |
+
#"https://github.com/open-mmlab/mmpose/blob/main/configs/body_3d_keypoint/pose_lift/h36m/pose-lift_simplebaseline3d_8xb64-200e_h36m.py",
|
39 |
#"https://download.openmmlab.com/mmpose/body3d/simple_baseline/simple3Dbaseline_h36m-f0ad73a4_20210419.pth") # pose3d="human3d"
|
40 |
#https://github.com/open-mmlab/mmpose/tree/main/configs/hand_2d_keypoint/topdown_regression
|
41 |
print("[INFO]: Downloaded models!")
|