Verah commited on
Commit
3c5981b
·
verified ·
1 Parent(s): 0e848c2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -1
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.