Skip to content

Commit 5c393b3

Browse files
Add notebook for DarkIR model (#2953)
[CVS-165152](https://jira.devtools.intel.com/browse/CVS-165152) --------- Co-authored-by: Andrei Kochin <andrei.kochin@intel.com>
1 parent 5a96702 commit 5c393b3

File tree

5 files changed

+1175
-0
lines changed

5 files changed

+1175
-0
lines changed

.ci/spellcheck/.pyspelling.wordlist.txt

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -155,6 +155,7 @@ CUDA
155155
CustomEncoderWav
156156
CVF
157157
CVPR
158+
CNNs
158159
Databricks
159160
databricks
160161
dataloader
@@ -164,6 +165,7 @@ DataLoaders
164165
DataModule
165166
dataset
166167
datasets
168+
DarkIR
167169
DaViT
168170
DCAE
169171
DDIM
@@ -436,6 +438,7 @@ Jupyter's
436438
JupyterLab
437439
Kaiming
438440
Kalman
441+
kagglehub
439442
kbps
440443
KD
441444
KDE
@@ -509,6 +512,10 @@ LogSoftmax
509512
LoRA
510513
LoRAs
511514
lora
515+
LOLBlur
516+
LOLv
517+
LOL
518+
LLIE
512519
lraspp
513520
LRASPP
514521
LTS
@@ -858,6 +865,7 @@ ResNet
858865
resnet
859866
RetinaFace
860867
RetroMAE
868+
Retinex
861869
RGB
862870
Riffusion
863871
riffusion
Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
# Low-Light Image Restoration with DarkIR model using OpenVINO™
2+
3+
[DarkIR](https://arxiv.org/abs/2412.13443) is an CNN model for Low-Light Image Restoration.
4+
5+
Photography during night or in dark conditions typically suffers from noise, low light and blurring issues due to the dim environment and the common use of long exposure. Although Deblurring and Low-light Image Enhancement (LLIE) are related under these conditions, most approaches in image restoration solve these tasks separately. DarkIR is an efficient and robust neural network for multi-task low-light image restoration. This approach proposes new attention mechanisms to enhance the receptive field of efficient CNNs. Current method reduces the computational costs in terms of parameters and MAC operations compared to previous methods. DarkIR achieves new state-of-the-art results on the popular LOLBlur, LOLv2 and Real-LOLBlur datasets, being able to generalize on real-world night and dark images.
6+
7+
![networks-scheme](https://raw.githubusercontent.com/cidautai/DarkIR/refs/heads/main/assets/networks-scheme.png)
8+
9+
[**image_source*](https://github.com/cidautai/DarkIR/blob/main/assets/networks-scheme.png)
10+
11+
You can find more information about this model in the [research paper](https://arxiv.org/abs/2412.13443) and GitHub [repository](https://github.com/cidautai/DarkIR).
12+
13+
14+
The notebook contains the following steps:
15+
16+
1. Prepare the PyTorch model.
17+
2. Prepare data and perform inference of the PyTorch model with image.
18+
3. Convert model to OpenVINO IR, using the model conversion API.
19+
4. Run inference with OpenVINO with image and video.
20+
5. Launch interactive demo
21+
22+
23+
## Installation Instructions
24+
25+
This is a self-contained example that relies solely on its own code.</br>
26+
We recommend running the notebook in a virtual environment. You only need a Jupyter server to start.
27+
For details, please refer to [Installation Guide](../../README.md)..
28+
29+
<img referrerpolicy="no-referrer-when-downgrade" src="https://static.scarf.sh/a.png?x-pxid=5b5a4db0-7875-4bfb-bdbd-01698b5b1a77&file=notebooks/darkir-image-restoration/README.md" />

0 commit comments

Comments
 (0)