xmrt commited on
Commit
b075423
1 Parent(s): 88152ab

3d change ND 2D NEW MODEL

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -2
  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==3.1.0"
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")#"https://github.com/open-mmlab/mmpose/blob/main/configs/body_3d_keypoint/pose_lift/h36m/pose-lift_simplebaseline3d_8xb64-200e_h36m.py",
 
 
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!")