|
|
@ -59,7 +59,7 @@ model = dict( |
|
|
|
frozen_stages=12, |
|
|
|
freeze_attn = True, |
|
|
|
freeze_ffn = True, |
|
|
|
samvit_checkpoint='/root/autodl-tmp/code/ViTPose/checkpoints/sam/sam_vit_b_01ec64.pth' |
|
|
|
samvit_checkpoint='/home/fhw/code/ViTPose/checkpoints/sam/sam_vit_b_01ec64.pth' |
|
|
|
), |
|
|
|
keypoint_head=dict( |
|
|
|
type='TopdownHeatmapSimpleHead', |
|
|
@ -79,7 +79,7 @@ model = dict( |
|
|
|
modulate_kernel=11, |
|
|
|
use_udp=True)) |
|
|
|
|
|
|
|
data_root = '/root/autodl-tmp/dataset/coco2017/' |
|
|
|
data_root = '/media/D/lxb/datasets/coco' |
|
|
|
|
|
|
|
data_cfg = dict( |
|
|
|
image_size=[192, 256], |
|
|
|