手把手教你安装Boundary-loss：Python与PyTorch环境配置完全指南

【免费下载链接】boundary-lossOfficial code for "Boundary loss for highly unbalanced segmentation", runner-up for best paper award at MIDL 2019. Extended version in MedIA, volume 67, January 2021.项目地址: https://gitcode.com/gh_mirrors/bo/boundary-loss

Boundary-loss是一个用于高度不平衡分割的开源项目，其官方代码来源于2019年MIDL最佳论文 runner-up，并在2021年1月的MedIA第67卷中发布了扩展版本。本指南将帮助你快速完成Python与PyTorch环境的配置，轻松上手Boundary-loss项目。

🌟 项目简介

Boundary-loss专注于解决高度不平衡分割问题，在医学影像分析等领域有着广泛的应用。它通过计算边界损失来提升分割效果，特别是在处理类别不平衡的数据时表现出色。项目提供了多种模型实现，如U-Net、Residual U-Net等，方便用户根据需求选择。

图：Boundary loss在ACDC四类别数据集上的分割结果，展示了其在多类别分割任务中的优势。

📋 环境要求

核心依赖

• Python 3.5+（推荐Python 3.9+以获得最佳兼容性）
• PyTorch 1.0+（推荐PyTorch 1.7+以支持最新功能）
• Scipy（任意版本）

实验复现额外依赖

• nibabel（用于3D体积切片处理）
• NumPy
• Matplotlib
• Scikit-image
• zsh（用于自动化脚本执行）

🚀 安装步骤

1️⃣ 克隆项目仓库

首先，将项目代码克隆到本地：

git clone https://gitcode.com/gh_mirrors/bo/boundary-loss cd boundary-loss

2️⃣ 创建虚拟环境（可选但推荐）

为了避免依赖冲突，建议使用虚拟环境：

# 使用conda创建虚拟环境 conda create -n boundary-loss python=3.9 conda activate boundary-loss # 或者使用venv python -m venv venv source venv/bin/activate # Linux/Mac venv\Scripts\activate # Windows

3️⃣ 安装PyTorch

根据你的系统和CUDA版本，从PyTorch官网获取安装命令。例如，对于CUDA 11.7：

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117

4️⃣ 安装其他依赖

使用pip安装所需的Python库：

pip install scipy numpy matplotlib scikit-image nibabel

⚙️ 验证安装

安装完成后，可以通过以下方式验证环境是否配置成功：

1. 检查Python版本：

python --version

2. 检查PyTorch版本：

import torch print(torch.__version__)

3. 尝试运行项目中的示例代码，例如查看损失函数定义：

cat losses.py

你可以在losses.py中找到BoundaryLoss的实现，位于第76行。

📝 使用说明

Boundary-loss的核心实现包括三个关键部分：

• 边界损失函数（BoundaryLoss）：位于losses.py#L76
• 距离图函数（one_hot2dist）：位于utils.py#L260
• 数据加载器转换（dist_map_transform）：位于dataloader.py#L105

图：Boundary loss与GDL在脑部影像分割任务中的效果对比，展示了Boundary loss在边界分割上的优势。

要使用Boundary-loss，你需要在数据加载过程中计算距离图，并在训练时将其与预测概率结合使用。项目提供了详细的使用示例，你可以参考main.py中的代码。

🤖 自动化实验

项目使用GNU Make来自动化实验流程，你可以通过以下命令运行预设的实验：

# 运行ISLES数据集实验 make -f isles.make # 运行WMH数据集实验 make -f wmh.make

这些命令将自动处理数据下载、预处理、训练和结果可视化等步骤，让你能够快速复现论文中的实验结果。

❓ 常见问题

损失函数可以为负数吗？

是的。由于距离图是有符号的（物体内部距离为负），完美的预测会导致负的损失值。在最小化损失的过程中，这不会产生问题。

需要归一化距离图吗？

这取决于具体的数据集。在项目的实验中没有进行归一化，但有用户反馈归一化在某些应用中有助于提升性能。

🎉 总结

通过本指南，你已经成功配置了Boundary-loss项目的Python与PyTorch环境。现在，你可以开始探索这个强大的分割工具，将其应用到自己的研究或项目中。如果在使用过程中遇到问题，可以参考项目的readme.md或查看代码中的详细注释。

祝你在分割任务中取得优异的成果！

【免费下载链接】boundary-lossOfficial code for "Boundary loss for highly unbalanced segmentation", runner-up for best paper award at MIDL 2019. Extended version in MedIA, volume 67, January 2021.项目地址: https://gitcode.com/gh_mirrors/bo/boundary-loss