
Vessel-WGAN-PyTorch: 采用生成对抗网络进行视网膜血管分割的研究
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:Vessel-WGAN-PyTorch项目利用生成对抗网络(WGAN)技术,在PyTorch框架下实现对视网膜血管图像的精确分割,推动了医学影像分析领域的进步。
船-维根-火炬
作者:谷玉超
该代码是使用Pytorch实现的。概述数据可以从服务器下载火车和测试数据。您也可以在eyedata文件夹中找到这些数据。
前处理:
该数据集包含20个训练图像,我的预处理的第一步是对它们进行随机裁剪至512 * 512大小。
第二步是随机调整火车图像的亮度、对比度以及色相。
我在代码中实现了此方法,因此可以方便地使用它。此外,基于GAN(生成对抗网络)的方法生成视网膜图像可以用作额外的数据源。
模型训练:
通过运行python train.py进行操作
依存关系
该代码依赖于以下库:Python 3.6 火炬皮尔结构体 vessel gan│├── eyedata # drive data│ ├── gycutils # 我的用于数据增强的工具包│ ├── Criterion.p
全部评论 (0)
还没有任何评论哟~


