Daniele Picone commited on
Commit ·
9b1191a
1
Parent(s): ef82f6a
Added models
Browse files- .gitattributes +8 -0
- README.md +44 -0
- dncnn3.pth +3 -0
- drunet_color.pth +3 -0
- drunet_deblocking_color.pth +3 -0
- drunet_deblocking_grayscale.pth +3 -0
- drunet_gray.pth +3 -0
- ffhq_10m.pt +3 -0
- ircnn_color.pth +3 -0
- ircnn_gray.pth +3 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,11 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
dncnn3.pth filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
drunet_color.pth filter=lfs diff=lfs merge=lfs -text
|
| 38 |
+
drunet_deblocking_color.pth filter=lfs diff=lfs merge=lfs -text
|
| 39 |
+
drunet_deblocking_grayscale.pth filter=lfs diff=lfs merge=lfs -text
|
| 40 |
+
drunet_gray.pth filter=lfs diff=lfs merge=lfs -text
|
| 41 |
+
ircnn_color.pth filter=lfs diff=lfs merge=lfs -text
|
| 42 |
+
ircnn_gray.pth filter=lfs diff=lfs merge=lfs -text
|
| 43 |
+
ffhq_10m.pt filter=lfs diff=lfs merge=lfs -text
|
README.md
CHANGED
|
@@ -1,3 +1,47 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
---
|
| 4 |
+
|
| 5 |
+
|
| 6 |
+
# Description
|
| 7 |
+
|
| 8 |
+
This repository contains models for image restoration collected from the following sources:
|
| 9 |
+
<https://github.com/cszn/IRCNN>
|
| 10 |
+
<https://github.com/cszn/DPIR>
|
| 11 |
+
<https://github.com/yuanzhi-zhu/DiffPIR>
|
| 12 |
+
|
| 13 |
+
|
| 14 |
+
|
| 15 |
+
# Citation
|
| 16 |
+
If you use this models, please cite:
|
| 17 |
+
|
| 18 |
+
```bibtex
|
| 19 |
+
@inproceedings{zhang2017learning,
|
| 20 |
+
title={Learning Deep CNN Denoiser Prior for Image Restoration},
|
| 21 |
+
author={Zhang, Kai and Zuo, Wangmeng and Gu, Shuhang and Zhang, Lei},
|
| 22 |
+
booktitle={IEEE Conference on Computer Vision and Pattern Recognition},
|
| 23 |
+
pages={3929--3938},
|
| 24 |
+
year={2017},
|
| 25 |
+
}
|
| 26 |
+
```
|
| 27 |
+
|
| 28 |
+
```bibtex
|
| 29 |
+
@article{zhang2021plug,
|
| 30 |
+
title={Plug-and-Play Image Restoration with Deep Denoiser Prior},
|
| 31 |
+
author={Zhang, Kai and Li, Yawei and Zuo, Wangmeng and Zhang, Lei and Van Gool, Luc and Timofte, Radu},
|
| 32 |
+
journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
|
| 33 |
+
volume={44},
|
| 34 |
+
number={10},
|
| 35 |
+
pages={6360-6376},
|
| 36 |
+
year={2021}
|
| 37 |
+
}
|
| 38 |
+
```
|
| 39 |
+
|
| 40 |
+
```bibtex
|
| 41 |
+
@inproceedings{zhu2023denoising, % DiffPIR
|
| 42 |
+
title={Denoising Diffusion Models for Plug-and-Play Image Restoration},
|
| 43 |
+
author={Yuanzhi Zhu and Kai Zhang and Jingyun Liang and Jiezhang Cao and Bihan Wen and Radu Timofte and Luc Van Gool},
|
| 44 |
+
booktitle={IEEE Conference on Computer Vision and Pattern Recognition Workshops (NTIRE)},
|
| 45 |
+
year={2023},
|
| 46 |
+
}
|
| 47 |
+
```
|
dncnn3.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cef4b4672a121b196a0525453b7091dc457a24b673105b38d623299f06fcc728
|
| 3 |
+
size 2668897
|
drunet_color.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:479abe3c5327dfd10ff54a80ec7d4098ca80752a5c9492cdff31cee430bec4b4
|
| 3 |
+
size 130579305
|
drunet_deblocking_color.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b26323e888230b1b5b58957add8dc9b9ea7fa2f69bdf6a0c3cca9db82167209f
|
| 3 |
+
size 130590513
|
drunet_deblocking_grayscale.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:af02bf265478dcc5cdbd8c92743599df0c8eea27aa5e9a0a9bd8d1387619a411
|
| 3 |
+
size 130580753
|
drunet_gray.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e27fb29456732c604c8ee3ac3e92ecadd2a7a4e36a8be675e92ebbf93a240de4
|
| 3 |
+
size 130569961
|
ffhq_10m.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:81d535743156ec6be34d8668e6920da94f0614074d7793a16c8fa9e306237faa
|
| 3 |
+
size 374417833
|
ircnn_color.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ce92ebb5991cb0bf13b9cf88d1e2ab64f2ef03103fc7af0a4b354641901767eb
|
| 3 |
+
size 18866429
|
ircnn_gray.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a5fb2ea80cf97d84cd539a6c576033c8daaaaee4a9dbf3e6aa9b6c4fcaf68dcf
|
| 3 |
+
size 18635829
|