From c3dd9b25c9443454c11fcbcb3647fad7d53cf5bf Mon Sep 17 00:00:00 2001 From: fhw Date: Wed, 10 Jul 2024 12:39:22 +0800 Subject: [PATCH] =?UTF-8?q?233=E6=9C=8D=E5=8A=A1=E5=99=A8=E9=85=8D?= =?UTF-8?q?=E7=BD=AE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../topdown_heatmap/coco/ViTSam_base_coco_256x192.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py b/configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py index 3f7cf86..1f8227e 100644 --- a/configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py +++ b/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, 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],