- 项目应用场景
-
- 面向医疗图像息肉分割场景,项目采用 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