
Matlab RGB2YCBCRN代码-SR-Pytorch-SRResNet:SR-火炬-SRResNet
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
这段代码是用于将RGB图像转换为YCbCrn格式,并应用于超分辨率任务中,使用了PyTorch框架下的SRResNet模型。
在PyTorch中实现论文《使用生成的对抗网络实现逼真的单图像超分辨率》中的SRResNet模型,并用以下命令训练该模型:
Matlab代码 `rgb2ycbcr` 转换为 PyTorch 版本。
用法:
运行 `main_srresnet.py`
可选参数包括:
- `-h`, `--help`: 显示帮助信息并退出。
- `--batchSize BATCHSIZE`: 训练批次大小。
- `--nEpochs NEPOCHS`: 要训练的周期数。
- `--lr LR`: 学习率,默认值为1e-4。
- `--step STEP`:
- `--cuda` : 使用CUDA进行GPU加速
- `--resume RESUME`
- `--start_epoch START_EPOCH`
- `--threads THREADS`
- `--pretrained PRETRAINED`
- `--vgg_loss`
- `--gpus GPUS`
这些参数可以根据需要调整以适应不同的训练需求。
全部评论 (0)
还没有任何评论哟~


