ai-forever
commited on
Commit
•
c091b84
1
Parent(s):
2d47e0e
Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,37 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# KandiSuperRes - diffusion model for 4K super resolution
|
2 |
+
|
3 |
+
[Habr Post](https://habr.com/ru/companies/sberbank/articles/775590/) | [Github](https://github.com/ai-forever/KandiSuperRes/) | [Telegram-bot](https://t.me/kandinsky21_bot) | [Technical Report](https://arxiv.org/pdf/2312.03511.pdf)| [Our text-to-image model](https://github.com/ai-forever/Kandinsky-3/tree/main)
|
4 |
+
|
5 |
+
![](assets/title.png)
|
6 |
+
|
7 |
+
## Description
|
8 |
+
|
9 |
+
KandiSuperRes is an open-source diffusion model for x4 super resolution. This model is based on the [Kandinsky 3.0](https://github.com/ai-forever/Kandinsky-3/tree/main) architecture with some modifications. For generation in 4K, the [MultiDiffusion](https://arxiv.org/pdf/2302.08113.pdf) algorithm was used, which allows to generate panoramic images. For more information: details of architecture and training, example of generations check out our [Habr post]().
|
10 |
+
|
11 |
+
## Installing
|
12 |
+
|
13 |
+
To install repo first one need to create conda environment:
|
14 |
+
|
15 |
+
```
|
16 |
+
conda create -n kandisuperres -y python=3.8;
|
17 |
+
source activate kandisuperres;
|
18 |
+
pip install -r requirements.txt;
|
19 |
+
```
|
20 |
+
|
21 |
+
## How to use
|
22 |
+
|
23 |
+
```python
|
24 |
+
from KandiSuperRes import get_SR_pipeline
|
25 |
+
from PIL import Image
|
26 |
+
|
27 |
+
sr_pipe = get_SR_pipeline(device='cuda', fp16=True)
|
28 |
+
|
29 |
+
lr_image = Image.open('')
|
30 |
+
sr_image = sr_pipe(lr_image)
|
31 |
+
```
|
32 |
+
|
33 |
+
## Authors
|
34 |
+
+ Anastasia Maltseva [Github](https://github.com/NastyaMittseva)
|
35 |
+
+ Vladimir Arkhipkin: [Github](https://github.com/oriBetelgeuse)
|
36 |
+
+ Andrey Kuznetsov: [Github](https://github.com/kuznetsoffandrey), [Blog](https://t.me/complete_ai)
|
37 |
+
+ Denis Dimitrov: [Github](https://github.com/denndimitrov), [Blog](https://t.me/dendi_math_ai)
|