医疗图像分割 | 基于Pyramid-Vision-Transformer算法实现医疗息肉分割

• 项目应用场景

• • 面向医疗图像息肉分割场景，项目采用 Pytorch + Pyramid-Vision-Transformer 深度学习算法来实现。

• 项目效果

• 项目细节 ==> 具体参见项目 README.md

• • (1) 模型架构

• • (2) 项目依赖，包括 python 3.8、pytorch 1.7.1、torchvision 0.8.2
  • (3) 下载处理好的数据集，下载预训练权重，具体下载方式可见 README，自取
  • (4) 执行训练

bash train.sh

• • (5) 执行推理

bash test.sh

• 项目获取

• • https://download.csdn.net/download/weixin_42405819/89099844