上传者: 42178688
|
上传时间: 2021-12-14 14:35:37
|
文件大小: 28.18MB
|
文件类型: -
船-维根-火炬
作者:谷玉超
电子邮件:
日期:2018-05-27
说明:该代码是的pytorch实现。
概述
数据
您可以从此服务器下载火车和测试数据。 您也可以在eyedata文件夹中找到数据。
前处理
该数据集包含20个训练图像,我的预处理的第一步是随机裁剪为512 * 512。 第二步是随机更改火车图像的亮度,对比度和色相。 我在代码中实现了此方法,因此可以方便地使用它。 此外,基于gan的生成视网膜图像的方法可以用作额外的数据源。
模型
训练
python train.py
如何使用
依存关系
此代码取决于以下库:
Python 3.6
火炬
皮尔
结构体
vessel gan
│
├── eyedata # drive data
│
├── gycutils # my utils for data augmentation
│
├── Criterion.p