Browse Source

233服务器配置

233
fhw 1 year ago
parent
commit
c3dd9b25c9
  1. 4
      configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py

4
configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py

@ -59,7 +59,7 @@ model = dict(
frozen_stages=12, frozen_stages=12,
freeze_attn = True, freeze_attn = True,
freeze_ffn = 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( keypoint_head=dict(
type='TopdownHeatmapSimpleHead', type='TopdownHeatmapSimpleHead',
@ -79,7 +79,7 @@ model = dict(
modulate_kernel=11, modulate_kernel=11,
use_udp=True)) use_udp=True))
data_root = '/root/autodl-tmp/dataset/coco2017/' data_root = '/media/D/lxb/datasets/coco'
data_cfg = dict( data_cfg = dict(
image_size=[192, 256], image_size=[192, 256],

Loading…
Cancel
Save