nicholasLane commited on
Commit
7755fe2
·
verified ·
1 Parent(s): 6446f47

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +37 -3
README.md CHANGED
@@ -1,3 +1,37 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ title: GRDFNet
3
+ colorFrom: red
4
+ colorTo: yellow
5
+ license: mit
6
+ short_description: A lightweight image restoration architecture
7
+ pinned: true
8
+ ---
9
+
10
+ # GRDFNet
11
+
12
+ GRDFNet is a lightweight image restoration network that combines gated and dilated residual blocks to deliver strong perceptual quality with modest compute requirements. The project ships with a Gradio application for interactive experimentation and can be accelerated further with TensorRT.
13
+
14
+ ## Recommended Configurations
15
+
16
+ - `num_sets = 3`, `feature_channels = 32`: strong quality while staying fast for most desktop workloads.
17
+ - `num_sets = 6`, `feature_channels = 48`: highest quality configuration; expect roughly a 4x slowdown versus the 32-channel model.
18
+ - `num_sets = 3`, `feature_channels = 24`: suggested for lightly compressed video inference; typically 50~75% faster than the 32-channel variant when deployed with TensorRT.
19
+
20
+ ## Performance Snapshot
21
+
22
+ Example TensorRT run on an NVIDIA RTX 4080 Super (16 GB):
23
+ ```
24
+ DEBUG: TensorRT initialized. Setting shape.
25
+ DEBUG: Shape set. Getting output shape.
26
+ [INFO] Input: 1280x720 -> 1280x720 -> ModelOut: 1280x720 @ 30000/1001 fps
27
+ DEBUG: Before NVENC initialization.
28
+ [prof] frames=209 avg=208.2 fps
29
+ [prof] frames=431 avg=215.1 fps
30
+ [prof] frames=654 avg=217.3 fps
31
+ [INFO] Processed 709 frames in 3.280s -> 216.2 FPS
32
+ ```
33
+
34
+ ## Resources
35
+
36
+ - Model weights: https://huggingface.co/nicholasLane/GRDFNet
37
+ - Hosted demo: https://huggingface.co/spaces/nicholasLane/GRDFNet