File size: 366 Bytes
f3748fb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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 |