(未完成)超分辨CVPR2023之LDL代码复现
文章贡献:GAN网络可以产生具有高感知质量的图像,但存在训练不稳定问题与伪影问题。分析了基于GAN的单个图像超分辨时视觉中的伪影是如何产生的,并针对GAN生成的伪影问题的方案:局部区域统计信息有助于伪影判别并进而生成mask引导训练过程。
根据README文件,先下载预训练模型和数据集,从蓝色字体点击会有很多种下载方式。
根据页面要求在目录中解压下载好的文件
对图像预处理
在script\data_preparation\datasets
文件夹中新建一个DIV2K的文件夹并下载DIV2K数据集(只需要训练集)放入其中并解压就会得到两个文件夹DIV2K_HR和DIV2K_train_LR_bicubic
首先要下载basicsr的依赖包在下载basicsr包
pip install numpy opencv-python lmdb pyyaml tb-nightly scipy torch torchvision scikit-image
pip install basicsr
然后打开extract_subimages.py
文件,根据注释释放处理图像所需要的参数
之后在配置好的环境中运行extract_subimages.py
文件,会得到下采样图片文件夹,终端中显示的MemoryError是内存问题引起的错误。