kirigaya/Slide-SAM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
transcendentsky 2024-03-20 21:43:27 +08:00
parent 635f319708
commit 88745c6105
4 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/vit_b.yaml
View File

@ -40,6 +40,8 @@ dataset:
data_txt_path: './datasets/dataset_list/' data_txt_path: './datasets/dataset_list/'
dataset2d_path: "/home1/quanquan/datasets/08_AbdomenCT-1K/" dataset2d_path: "/home1/quanquan/datasets/08_AbdomenCT-1K/"
cache_data_path: '/home1/quanquan/datasets/cached_dataset2/' cache_data_path: '/home1/quanquan/datasets/cached_dataset2/'
cache_prefix: ['6016'] # '07'
specific_label: [2]
# sam_checkpoint: "/quanquan/code/projects/medical-guangdong/segment-anything/sam_vit_b_01ec64.pth" # 103 server # sam_checkpoint: "/quanquan/code/projects/medical-guangdong/segment-anything/sam_vit_b_01ec64.pth" # 103 server
# model_type: "vit_b" # model_type: "vit_b"

2
configs/vit_sub.yaml
View File

@ -34,7 +34,7 @@ dataset:
types: ['3d'] # ['3d', '2d'] types: ['3d'] # ['3d', '2d']
split: 'train' split: 'train'
data_root_path: '/home1/quanquan/datasets/' data_root_path: '/home1/quanquan/datasets/'
dataset_list: ["pancreas"] dataset_list: ["example"] # for example_train.txt
data_txt_path: './datasets/dataset_list/' data_txt_path: './datasets/dataset_list/'
dataset2d_path: "/home1/quanquan/datasets/08_AbdomenCT-1K/" dataset2d_path: "/home1/quanquan/datasets/08_AbdomenCT-1K/"
cache_data_path: '/home1/quanquan/datasets/cached_dataset2/' cache_data_path: '/home1/quanquan/datasets/cached_dataset2/'

2
core/ddp_sub.py
View File

@ -105,7 +105,7 @@ if __name__ == "__main__":
world_size = n_gpus world_size = n_gpus
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--config", default="./configs/vit_sub_rectum.yaml") parser.add_argument("--config", default="./configs/vit_sub.yaml")
parser.add_argument("--func", default="train") parser.add_argument("--func", default="train")
parser.add_argument("--reuse", action="store_true") parser.add_argument("--reuse", action="store_true")

1
core/lora_sam.py
View File

@ -33,6 +33,7 @@ class _LoRA_qkv(nn.Module):
self.linear_a_v = linear_a_v self.linear_a_v = linear_a_v
self.linear_b_v = linear_b_v self.linear_b_v = linear_b_v
self.dim = qkv.in_features self.dim = qkv.in_features
self.in_features = qkv.in_features
self.w_identity = torch.eye(qkv.in_features) self.w_identity = torch.eye(qkv.in_features)
def forward(self, x): def forward(self, x):