Miroslav Purkrabek commited on
Commit
b42544a
·
1 Parent(s): 28225b6

load models upon each run

Browse files
Files changed (2) hide show
  1. README.md +0 -7
  2. configs/bmp_D3.yaml +2 -2
README.md CHANGED
@@ -7,11 +7,6 @@ tags:
7
  - person-detection
8
  - human-pose-estimation
9
  - instance-segmentation
10
- datasets:
11
- - COCO
12
- - MPII
13
- - AIC
14
- - OCHuman
15
  models:
16
  - vrg-prague/BBoxMaskPose
17
  - facebook/sam2.1-hiera-base-plus
@@ -23,8 +18,6 @@ sdk: gradio
23
  sdk_version: 5.33.0
24
  app_file: app.py
25
  pinned: false
26
- preload_from_hub:
27
- - vrg-prague/BBoxMaskPose MaskPose-b.pth,rtmdet-ins-l-mask.pth
28
  license: gpl-3.0
29
  short_description: '[ICCV 25] Image Demo for BBoxMaskPose method'
30
  ---
 
7
  - person-detection
8
  - human-pose-estimation
9
  - instance-segmentation
 
 
 
 
 
10
  models:
11
  - vrg-prague/BBoxMaskPose
12
  - facebook/sam2.1-hiera-base-plus
 
18
  sdk_version: 5.33.0
19
  app_file: app.py
20
  pinned: false
 
 
21
  license: gpl-3.0
22
  short_description: '[ICCV 25] Image Demo for BBoxMaskPose method'
23
  ---
configs/bmp_D3.yaml CHANGED
@@ -4,7 +4,7 @@
4
  # This configuration is good for the BMP loop as was used for most of the experiments.
5
  detector:
6
  det_config: 'mmpose/configs/mmdet/rtmdet/rtmdet-ins_l_8xb32-300e_coco.py'
7
- det_checkpoint: 'rtmdet-ins-l-mask.pth'
8
 
9
  # Detectors D and D' could be different.
10
  det_prime_config: null
@@ -12,7 +12,7 @@ detector:
12
 
13
  pose_estimator:
14
  pose_config: 'mmpose/configs/MaskPose/ViTb-multi_mask.py'
15
- pose_checkpoint: 'MaskPose-b.pth'
16
 
17
  sam2:
18
  sam2_config: 'configs/samurai/sam2.1_hiera_b+.yaml' # Use SAMURAI as it has img_size 1024 (SAM-2.1 has 512)
 
4
  # This configuration is good for the BMP loop as was used for most of the experiments.
5
  detector:
6
  det_config: 'mmpose/configs/mmdet/rtmdet/rtmdet-ins_l_8xb32-300e_coco.py'
7
+ det_checkpoint: 'https://huggingface.co/vrg-prague/BBoxMaskPose/resolve/main/rtmdet-ins-l-mask.pth'
8
 
9
  # Detectors D and D' could be different.
10
  det_prime_config: null
 
12
 
13
  pose_estimator:
14
  pose_config: 'mmpose/configs/MaskPose/ViTb-multi_mask.py'
15
+ pose_checkpoint: 'https://huggingface.co/vrg-prague/BBoxMaskPose/resolve/main/MaskPose-b.pth'
16
 
17
  sam2:
18
  sam2_config: 'configs/samurai/sam2.1_hiera_b+.yaml' # Use SAMURAI as it has img_size 1024 (SAM-2.1 has 512)