# RED-Net
This repository is implementation of the "Image Restoration Using Very Deep Convolutional Encoder-Decoder Networks with Symmetric Skip Connections". <br />
To reduce computational cost, it adopts stride 2 for the first convolution layer and the last transposed convolution layer.
## Requirements
- PyTorch
- tqdm
- Numpy
- Pillow
## Results
sigma=10,REDNet10:
![07_plt_x[10]_REDNet10](./data/07_plt_x[10]_REDNet10.png)
![08_plt_x[10]_REDNet10](./data/08_plt_x[10]_REDNet10.png)
## Usages
### Train
```bash
python main.py --arch "REDNet30" \ # REDNet10, REDNet20, REDNet30
--images_dir "" \
--outputs_dir "" \
--gaussian_noise_level 10 \
--patch_size 50 \
--batch_size 16 \
--num_epochs 1000 \
--lr 1e-4 \
--threads 8 \
--seed 123 \
```
### Test
```bash
python example --arch "REDNet30" \ # REDNet10, REDNet20, REDNet30
--weights_path "" \
--image_path "" \
--outputs_dir "" \
--gaussian_noise_level 10
```
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
资源配套文章:https://blog.csdn.net/qq_36584673/article/details/141996345 data:去噪后图像结果保存位置 datasets:数据集所在文件夹 Plt:训练过程指标曲线可视化位置(Loss、PSNR、SSIM与Epoch关系曲线) weights:训练模型保存位置 dataset.py:封装数据集 draw_evaluation.py:绘制指标曲线 example.py:测试Noise2Void main.py:训练Noise2Void model.py:模型实现(U-Net) utils.py:工具类 使用方式:见配套文章(包含完整的模型训练验证测试流程,非常详细的使用说明,轻松跑起来;模型算法讲解,代码复现思路,注释清晰;总结反思、创新思路;结果展示等)
资源推荐
资源详情
资源评论
收起资源包目录
Noise2Void-pytorch-master.zip (20个子文件)
Noise2Void-pytorch-master
utils.py 7KB
weights
best_N2V_Unet_rgb_[30].pth 2.68MB
best_N2V_Unet_gray_[25].pth 2.67MB
main.py 10KB
data
dataset.py 22KB
model.py 9KB
.idea
workspace.xml 2KB
misc.xml 266B
REDNet-pytorch-master.iml 316B
inspectionProfiles
profiles_settings.xml 174B
modules.xml 301B
.gitignore 50B
datasets
Plt
evalution_plt_[30]_N2V_Unet.png 427KB
draw_evaluation.py 3KB
example.py 8KB
.gitignore 16B
__pycache__
model.cpython-38.pyc 5KB
utils.cpython-38.pyc 6KB
dataset.cpython-38.pyc 6KB
README.md 1KB
共 20 条
- 1
资源评论
- 我为河财扛大旗2024-09-13你好作者,dataset.py文件中424行Unresolved reference 'noisy_image'错误求解决
- m0_751013232024-10-05资源值得借鉴的内容很多,那就浅学一下吧,值得下载!
十小大
- 粉丝: 1w+
- 资源: 1528
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功