| input_size: [3, 112, 112] | |
| color_space: 'RGB' | |
| name: 'base' | |
| output_dim: 512 | |
| start_from: '' | |
| freeze: False | |
| mask_ratio: 0.0 | |
| rpe_config: | |
| name: KPRPE_shared | |
| rpe_on: k | |
| shared_head: True | |
| mode: ctx | |
| method: product | |
| ratio: 1.9 | |
| ctx_type: 'rel_keypoint_splithead_unshared' | |
| num_keypoints: 5 | |
| yaml_path: models/vit_kprpe/configs/v1_base_kprpe_splithead_unshared.yaml |