自己做网站软件,百度地图人工服务,建网站用营业执照吗,泰州网站制作方案定制一、下载数据集及项目包
1. 数据集
1.1 文件夹框架的介绍#xff0c;如下图所示#xff1a;主要有train和val#xff0c;分别有高清#xff08;HR#xff09;和低清#xff08;LR#xff09;的图像。 1.2 原图先通过分割尺寸的脚本先将数据集图片处理成两个相同的图像…一、下载数据集及项目包
1. 数据集
1.1 文件夹框架的介绍如下图所示主要有train和val分别有高清HR和低清LR的图像。 1.2 原图先通过分割尺寸的脚本先将数据集图片处理成两个相同的图像组HR和LR。
如训练x4的ESRGAN模型那么我们需要将HR的图像尺寸与LR的图像尺寸比例是4:1。在我的训练中我将HR的图像尺寸分割成了480x480LR的图像分割成了120x120。如下图所示。 随后将分割好的图像按照train和val的分类分成如1.1图中的文件结构。 2. 项目包
在我的下载资源中有SR项目包的下载后续上链接。也可以在我上传的资源中下载。
二、训练ESRGAN
ESRGAN模型包括生成模型的训练和判别模型的训练。
2.1 配置RRDBNet_train.py生成模型的参数及训练
2.1.1 训练的图像路径设置dataroot_gt为HR图像的路径、dataroot_lq为LR图像的路径。 2.1.2 batch_size_per_gpu为batchsize的设置根据显存大小相应设置显存越大可以设置的值越大但是训练时间也会增大。 2.1.3 val的数据集路径设置dataroot_gt为HR的图像路径、dataroot_lq为LR图像的路径。 2.1.4 训练迭代次数的设置可以设置到10万或者更大 2.1.5 训练结果指标的计算psnr和ssim。val_freq参数为保存结果的频率。下图中我的设置为1e3即1000轮保存一次。 2.1.6 保存训练权重的频率设置。下图中我的设置为1e3即为1000次保存一次训练权重。 2.1.7 RRDBNet_train.py的训练
python basicsr/train.py -opt options\train\ESRGAN\train_RRDBNet_PSNR_x4.yml
2.2 配置ESRGAN_train.py判别模型的参数及训练
2.2.1 ESRGAN_train.py的参数设置
ESRGAN_train.py的参数设置与RRDBNet_train.py相同但是多了一个pretrain_network_g参数的设置即填RRDBNet_train.py训练完以后最好的那次权重路径。 2.2.2 ESRGAN_train.py的训练
python basicsr/train.py -opt options\train\ESRGAN\train_RRDBNet_PSNR_x4.yml
三、测试
3.1 测试图片路径的设置
包括HR和LR的路径分别为dataroot_gt和dataroot_lq。 3.2 ESRGAN模型权重的路径导入
在pretrain_network_g参数中导入ESRGAN模型训练完后生成的权重路径。 四、训练中断后继续训练
只需要在训练代码后加上--auto_resume
python basicsr/train.py -opt options\train\ESRGAN\train_RRDBNet_PSNR_x4.yml --auto_resume ------------------ 今天不学习明天变垃圾。 ---------------------
