Update README.md
Browse files
README.md
CHANGED
|
@@ -6,9 +6,12 @@ tags:
|
|
| 6 |
|
| 7 |
# Intro
|
| 8 |
These are my efforts to train a real-world usable [Cascaded Gaze](https://github.com/Ascend-Research/CascadedGaze) image denoising network.
|
| 9 |
-
|
| 10 |
denoise_util.py includes all definitions required to use Cascaded Gaze networks with PyTorch.
|
| 11 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
# Models
|
| 13 |
|
| 14 |
**small** (`cg_denoise_jpg+webp_artifacts_small.safetensors`)
|
|
@@ -44,6 +47,12 @@ model.requires_grad_(False)
|
|
| 44 |
model.eval()
|
| 45 |
```
|
| 46 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
**v1**
|
| 48 |
- an early experiment, not recommended
|
| 49 |
- ~ 132M params, trained on 256 * 256 **RGB** patches for intermediate jpg & webp compression artefact removal. It's been trained on about 700k samples (photographs only) at a precision of bf16. Also capable of removing ISO-like noise and gaussian noise.
|
|
|
|
| 6 |
|
| 7 |
# Intro
|
| 8 |
These are my efforts to train a real-world usable [Cascaded Gaze](https://github.com/Ascend-Research/CascadedGaze) image denoising network.
|
|
|
|
| 9 |
denoise_util.py includes all definitions required to use Cascaded Gaze networks with PyTorch.
|
| 10 |
|
| 11 |
+
You can find inference code for some of these models on my github:
|
| 12 |
+
|
| 13 |
+
- [ComfyUI custom nodes](https://github.com/crimro-se/ComfyUI-CascadedGaze)
|
| 14 |
+
|
| 15 |
# Models
|
| 16 |
|
| 17 |
**small** (`cg_denoise_jpg+webp_artifacts_small.safetensors`)
|
|
|
|
| 47 |
model.eval()
|
| 48 |
```
|
| 49 |
|
| 50 |
+
**sidd** (`cg_sidd.safetensors`)
|
| 51 |
+
The official SIDD benchmark trained CascadedGaze model. I have ported the weights and added metadata such that it can be used as easily as my small model.
|
| 52 |
+
|
| 53 |
+
However, my view is that the SIDD dataset is poor, and as a result this model is not useful in any task.
|
| 54 |
+
|
| 55 |
+
|
| 56 |
**v1**
|
| 57 |
- an early experiment, not recommended
|
| 58 |
- ~ 132M params, trained on 256 * 256 **RGB** patches for intermediate jpg & webp compression artefact removal. It's been trained on about 700k samples (photographs only) at a precision of bf16. Also capable of removing ISO-like noise and gaussian noise.
|