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

try to pre-load models in build time

Browse files
Files changed (1) hide show
  1. configs/bmp_D3.yaml +2 -2
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: '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,7 +12,7 @@ detector:
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)
 
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
 
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)