1、下载官方权重
官方的预训练权重:https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth
下载后保存在:work_dir/SAM/sam_vit_b_01ec64.pth 目录
2、摆放数据集
因为MedSAM 分割模型需要对3D数据集进行切片处理,也就是对nii.gz 数据处理成 npy 格式
具体的MedSAM项目参考:第一章:MedSAM 视觉大模型介绍_medsam 系统简介-CSDN博客
这里只需要将数据集的目录名称在代码修改一下即可
需要把数据放在 data 目录下
这里训练的数据集目录如下: