finetune-SAM

finetune-SAM copied to clipboard

the poor performance when i adjust the args.out_size=512

6

  Previously, I used the code from the Medical-SAM-Adapter project (https://github.com/KidsWithTokens/Medical-SAM-Adapter) to perform multi-organ segmentation on the Synapse dataset, achieving an average Dice accuracy of 89%. However, after adjusting...

ShChen233

Hi， I want to fine-tune the SAM mask decoder to only a single labeled dataset unsing supervised training (i.e., Setup 1 in your paper). Since I have no experience in...

happyday521

Self-supervised training

1

Is there code for self-supervised training? If not, can you provide reference code?

ShijianRuan

How to prepare a dataset for multi-cls segmentation?

1

How to prepare a dataset for multi-cls segmentation? Thank you!

itbear-shu

CUDA not found

1

I am getting the following error when I try to run train_singlegpu_demo.sh File "/home/mayank/Projects/finetune-SAM/SingleGPU_train_finetune_noprompt.py", line 212, in train_model(trainloader,valloader,args.dir_checkpoint,args.epochs) File "/home/mayank/Projects/finetune-SAM/SingleGPU_train_finetune_noprompt.py", line 51, in train_model sam = sam_model_registry[args.arch](args,checkpoint=os.path.join(args.sam_ckpt),num_classes=args.num_cls) File "/home/mayank/Projects/finetune-SAM/models/sam/build_sam.py", line...

jmayank23

ddp train failed

3

when i use ddp_splitgpu_train_finetune_noprompt.py , the error raised, “all tensors to be on the same device,but found at least two devices,cuda:1 and cuda:0, how can i solve it.

ZHIZIHUABU

GPU allocated

1

Thank you for your excellent work on the paper. I encountered a CUDA out-of-memory error with a batch size of 2 on an A100 GPU. I noticed in Table 3...

rykawamura