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!")
|