From 82c6dcc204db8dec4ddc321141d4d646e53c1a62 Mon Sep 17 00:00:00 2001 From: shishaoyu Date: Mon, 1 Apr 2024 08:04:03 +0000 Subject: [PATCH 1/6] update PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/train_multi_npus.sh. Signed-off-by: shishaoyu --- .../MedSAM_for_PyTorch/train_multi_npus.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/train_multi_npus.sh b/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/train_multi_npus.sh index 6e7fe29fcc..010b47aed4 100644 --- a/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/train_multi_npus.sh +++ b/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/train_multi_npus.sh @@ -11,8 +11,8 @@ MASTER_PORT=6789 python train_multi_npus.py \ --task_name "${TASK}" \ --model_type vit_b \ - --tr_npy_path ../datasets/npy/CT_Abd \ - --checkpoint ../models/sam_vit_b_01ec64.pth \ + --tr_npy_path ./datasets/npy/CT_Abd \ + --checkpoint ./models/sam_vit_b_01ec64.pth \ --work_dir ./work_dir \ --num_epochs 100 \ --batch_size 4 \ -- Gitee From 5b4d4b77bd294dd6cc37748143a38a65e4ee2c93 Mon Sep 17 00:00:00 2001 From: shishaoyu Date: Mon, 1 Apr 2024 08:06:06 +0000 Subject: [PATCH 2/6] update PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/README.md. Signed-off-by: shishaoyu --- .../cv/semantic_segmentation/MedSAM_for_PyTorch/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/README.md b/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/README.md index dedf12cb46..10c025535b 100644 --- a/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/README.md +++ b/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/README.md @@ -28,7 +28,7 @@ ``` url=https://gitee.com/ascend/ModelZoo-PyTorch.git - code_path=PyTorch/contrib/cv/semantic_segmentation/ + code_path=PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch ``` # 准备训练环境 -- Gitee From 0228e19ee65681ab18c93dfaf3e42f15ca51dc4a Mon Sep 17 00:00:00 2001 From: shishaoyu Date: Mon, 1 Apr 2024 08:06:24 +0000 Subject: [PATCH 3/6] update PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/requirements.txt. Signed-off-by: shishaoyu --- .../semantic_segmentation/MedSAM_for_PyTorch/requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/requirements.txt b/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/requirements.txt index dc1088606a..229342c8de 100644 --- a/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/requirements.txt +++ b/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/requirements.txt @@ -1,4 +1,4 @@ -torchvision +torchvision==0.15.2 opencv-python pycocotools matplotlib -- Gitee From 09f4f1ce9a6bfd880e2b31dfa78d6c9159002f6c Mon Sep 17 00:00:00 2001 From: shishaoyu Date: Mon, 1 Apr 2024 08:10:12 +0000 Subject: [PATCH 4/6] update PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/train_multi_npus.sh. Signed-off-by: shishaoyu --- .../MedSAM_for_PyTorch/train_multi_npus.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/train_multi_npus.sh b/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/train_multi_npus.sh index 010b47aed4..f2c6c08112 100644 --- a/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/train_multi_npus.sh +++ b/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/train_multi_npus.sh @@ -11,7 +11,7 @@ MASTER_PORT=6789 python train_multi_npus.py \ --task_name "${TASK}" \ --model_type vit_b \ - --tr_npy_path ./datasets/npy/CT_Abd \ + --tr_npy_path ./data/npy/CT_Abd \ --checkpoint ./models/sam_vit_b_01ec64.pth \ --work_dir ./work_dir \ --num_epochs 100 \ -- Gitee From dc743beb8be9464d35bb787c19e17d49edb490ff Mon Sep 17 00:00:00 2001 From: shishaoyu Date: Mon, 1 Apr 2024 08:42:02 +0000 Subject: [PATCH 5/6] update PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/eval_multi_npus.sh. Signed-off-by: shishaoyu --- .../MedSAM_for_PyTorch/eval_multi_npus.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/eval_multi_npus.sh b/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/eval_multi_npus.sh index 375e3c3e47..c259804e33 100644 --- a/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/eval_multi_npus.sh +++ b/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/eval_multi_npus.sh @@ -11,8 +11,8 @@ MASTER_PORT=6789 python eval_multi_npus.py \ --task_name "${TASK}" \ --model_type vit_b \ - --data_path ../datasets/npy/CT_Abd \ - --checkpoint ../models/sam_vit_b_01ec64.pth \ + --data_path ./datasets/npy/CT_Abd \ + --checkpoint ./models/sam_vit_b_01ec64.pth \ --work_dir ./work_dir \ --num_workers 8 \ --nnodes "${NNODES}" \ -- Gitee From fd39a61a850f5763a60c8ae00918adb9d721f0a7 Mon Sep 17 00:00:00 2001 From: shishaoyu Date: Mon, 1 Apr 2024 08:42:49 +0000 Subject: [PATCH 6/6] update PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/README.md. Signed-off-by: shishaoyu --- .../cv/semantic_segmentation/MedSAM_for_PyTorch/README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/README.md b/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/README.md index 10c025535b..ca1c2f02a9 100644 --- a/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/README.md +++ b/PyTorch/contrib/cv/semantic_segmentation/MedSAM_for_PyTorch/README.md @@ -155,11 +155,11 @@ ```shell # 原始的SAM模型权重路径 - sam_ckpt_path = "../models/sam_vit_l_0b3195.pth" + sam_ckpt_path = "./models/sam_vit_l_0b3195.pth" # 训练好的MedSAM模型权重路径 medsam_ckpt_path = "workdir/MedSAM-ViT-L-20230912-1734/medsam_model_latest.pth" # 转换后输出的模型权重路径 - save_path = "../models/medsam_vit_l_train.pth" + save_path = "./models/medsam_vit_l_train.pth" ``` 然后执行转换脚本得到转换后的模型： ```shell -- Gitee