jaychan0 / fine_tune_sam_mis

This is a UI project, which uses medical image to fine tune SAM in hugging face structure.

fine_tune_SAM_MIS

This is a UI project, which uses medical images to fine tune SAM in hugging face structure. You can choose the checkpoint in option module,including lung, heart, colon, spleen and original vit_b. Besides, you can also use your own checkpoint through the option module. The size of images and GT must be 256*256

You can see the code in the branches "master"

download the checkpoint

If you want to download the checkpoint, please click following link

link：https://pan.baidu.com/s/13mc85Qt1qX4RM3KdfILy-w

code：czr0

make sure the folder be in the same file directory with the code

the file structure will be:

--res_checkpiont/

--MainWindowShow.py

--UI_329_up_up.py

display of effect drawing

original vit_b in spleen image segment

fine tune checkpoint in spleen image segment

fine_tune_sam_mis's Issues

您好，百度网盘地址失效了，可以更新一下吗