扩散模型在图像生成中的应用实践
数据加载模块结构 ├── 核心接口 │ └── torch.utils.data.Dataset │ ├── len() # 数据集大小 │ └── getitem() # 数据采样 ├── 数据集实现 │ ├── BRATSDataset3D (bratsloader.py) │ │ ├── 数据特征:3D医学图像(nii.gz格式) │ │ ├── 目录结构要求: │ │ │ └── 直接包含nii文件(无子目录) │ │ │ ├── brats_xxx_t1.nii.gz │ │ │ ├── brats_xxx_t1ce.nii.gz │ │ │ └── ...(多模态数据) │ │ └── 切片处理:将3D数据切片为2D(155 slices/volume) │ │ │ ├── ISICDataset (isicloader.py) │ │ ├── 数据特征:皮肤镜图像(jpg + png掩码) │ │ ├── 目录结构要求: │ │ │ ├── ISBI2016_ISIC_Part3B__GroundTruth.csv │ │ │ ├── 图像文件(jpg) │ │ │ └── 掩码文件(png)
│ │ │ └── CustomDataset (custom_dataset_loader.py) │ ├── 数据特征:通用分割数据(png格式) │ └── 目录结构要求: │ ├── images/.png │ └── masks/.png ├── 数据变换 │ └── torchvision.transforms │ ├── Resize() # 统一图像尺寸 │ ├── ToTensor() # 张量转换 │ └── Compose() # 组合变换 └── 数据加载器 └── torch.utils.data.DataLoa
1