数据集配置:
当前目录下新建文件夹
Datasets
目录名分别为
SYSUMM01、RegDB、LLCM。
环境配置：
安装了python3.9，pytorch用的最新版本，支持cuda12.4
bash
conda install pytorch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 pytorch-cuda=12.4 -c pytorch -c nvidia
值得注意的点：
其中使用的Pillow有用到ANTIALIAS方法，最新的Pillow已经不支持，需要将data_loader中的ANTIALIAS 替换为LANCZOS.