MedLSAM 项目使用教程

目录结构及介绍

MedLSAM 项目的目录结构如下：

MedLSAM/
├── checkpoint/
│   ├── medlam.pth
│   ├── sam_vit_b_01ec64.pth
│   └── medsam_vit_b.pth
├── config/
│   └── test_config/
│       ├── test_structseg_medlam_wpl_medsam.txt
│       └── test_structseg_medlam_spl_sam.txt
├── MedLSAM_SPL_Inference.py
├── MedLSAM_WPL_Inference.py
├── README.md
└── requirements.txt

目录介绍

• checkpoint/: 存放预训练模型的检查点文件。
• config/: 存放配置文件，用于不同任务的配置。
• MedLSAM_SPL_Inference.py 和 MedLSAM_WPL_Inference.py: 项目的启动文件，用于执行推理任务。
• README.md: 项目说明文档。
• requirements.txt: 项目依赖文件。

项目的启动文件介绍

MedLSAM_SPL_Inference.py

该文件用于执行基于子补丁定位（SPL）的推理任务。使用方法如下：

CUDA_VISIBLE_DEVICES=0 python MedLSAM_SPL_Inference.py --config_file path/to/your/test_medlsam_config.txt

MedLSAM_WPL_Inference.py

该文件用于执行基于加权补丁定位（WPL）的推理任务。使用方法如下：

CUDA_VISIBLE_DEVICES=0 python MedLSAM_WPL_Inference.py --config_file path/to/your/test_medlsam_config.txt

项目的配置文件介绍

配置文件位于 config/test_config/ 目录下，包含以下文件：

• test_structseg_medlam_wpl_medsam.txt: 用于结构化分割任务的配置文件。
• test_structseg_medlam_spl_sam.txt: 用于子补丁定位任务的配置文件。

配置文件示例

[DEFAULT]
model_path = checkpoint/medlam.pth
input_data_path = path/to/your/input_data
output_data_path = path/to/your/output_data

配置文件中包含模型路径、输入数据路径和输出数据路径等参数，用户可以根据需要进行修改。

以上是 MedLSAM 项目的基本使用教程，包括项目的目录结构、启动文件和配置文件的介绍。希望对您有所帮助！