Hugging Face's logo

Spaces:
NN-BRD
/
MMpose
Build error

App Files Community
xmrt commited on
Commit
a55b892
•
1 Parent(s): 9fd87a1

mmdet again

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -1
  2. main_noweb.py +1 -1
Dockerfile CHANGED
@@ -17,7 +17,8 @@ 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
 
22
  RUN git clone https://github.com/open-mmlab/mmpose.git
23
 
 
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
 
main_noweb.py CHANGED
@@ -30,7 +30,7 @@ else:
30
  os.system("nvidia-smi")
31
 
32
  print("[INFO]: Imported modules!")
33
- human = MMPoseInferencer(pose2d="human", pose2d_weights="https://download.openmmlab.com/mmpose/v1/body_2d_keypoint/simcc/coco/simcc_mobilenetv2_wo-deconv-8xb64-210e_coco-256x192-4b0703bb_20221010.pth")
34
  hand = MMPoseInferencer("hand")
35
  human3d = MMPoseInferencer(pose3d="human3d")
36
  track_model = YOLO('yolov8n.pt') # Load an official Detect model
 
30
  os.system("nvidia-smi")
31
 
32
  print("[INFO]: Imported modules!")
33
+ human = MMPoseInferencer(pose2d="human")
34
  hand = MMPoseInferencer("hand")
35
  human3d = MMPoseInferencer(pose3d="human3d")
36
  track_model = YOLO('yolov8n.pt') # Load an official Detect model