Spaces:
Running
on
Zero
Running
on
Zero
Miroslav Purkrabek
commited on
Commit
·
b42544a
1
Parent(s):
28225b6
load models upon each run
Browse files- README.md +0 -7
- 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)
|