MedSAM 学习笔记（续）：训练自定义数据集

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 目录下

这里训练的数据集目录如下：