Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
DiffBIR_pytorch
Commits
07447674
Commit
07447674
authored
Jul 12, 2024
by
bailuo
Browse files
add SCNet add icon
parent
46ea1a27
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
12 additions
and
6 deletions
+12
-6
README.md
README.md
+12
-6
icon.png
icon.png
+0
-0
No files found.
README.md
View file @
07447674
...
@@ -64,14 +64,16 @@ Tips:以上dtk驱动、python、torch等DCU相关工具版本需要严格一
...
@@ -64,14 +64,16 @@ Tips:以上dtk驱动、python、torch等DCU相关工具版本需要严格一
## 数据集
## 数据集
下载地址(训练+测试集):https://www.image-net.org/ (imagenet1k)
下载地址(训练+测试集):https://www.image-net.org/ (imagenet1k)
\
或者从SCNet下载
[
imagenet1k
](
http://113.200.138.88:18080/aidatasets/project-dependency/imagenet-2012
)
datasets
datasets
|- train
|- train
|- n01440764
|- n01440764
|- xxx.JPEG
|- xxx.JPEG
下载地址(训练+测试集):https://www.robots.ox.ac.uk/~vgg/data/pets/
下载地址(训练+测试集):https://www.robots.ox.ac.uk/~vgg/data/pets/
\
或者从SCNet下载
[
vgg
](
http://113.200.138.88:18080/aidatasets/project-dependency/vgg
)
datasets
datasets
|- images
|- images
...
@@ -107,7 +109,8 @@ Tips:以上dtk驱动、python、torch等DCU相关工具版本需要严格一
...
@@ -107,7 +109,8 @@ Tips:以上dtk驱动、python、torch等DCU相关工具版本需要严格一
### 阶段二
### 阶段二
1、模型准备(Stable Diffusion v2.1): https://huggingface.co/stabilityai/stable-diffusion-2-1-base/resolve/main/v2-1_512-ema-pruned.ckpt
1、模型准备(Stable Diffusion v2.1): https://huggingface.co/stabilityai/stable-diffusion-2-1-base/resolve/main/v2-1_512-ema-pruned.ckpt
\
或者从SCNet下载
[
stable-diffusion-2-1-base
](
http://113.200.138.88:18080/aimodels/stable-diffusion-2-1-base/-/raw/main/v2-1_512-ema-pruned.ckpt?ref_type=heads&inline=false
)
2、初始化模型参数
2、初始化模型参数
...
@@ -131,9 +134,11 @@ Tips:以上dtk驱动、python、torch等DCU相关工具版本需要严格一
...
@@ -131,9 +134,11 @@ Tips:以上dtk驱动、python、torch等DCU相关工具版本需要严格一
### general Image
### general Image
模型下载地址:
模型下载地址:
*
https://huggingface.co/lxq007/DiffBIR/resolve/main/general_full_v1.ckpt
*
https://huggingface.co/lxq007/DiffBIR/resolve/main/general_full_v1.ckpt
\
或者从SCNet下载
[
general_full_v1
](
http://113.200.138.88:18080/aimodels/DiffBIR/-/raw/main/general_full_v1.ckpt?ref_type=heads&inline=false
)
*
https://huggingface.co/lxq007/DiffBIR/resolve/main/general_swinir_v1.ckpt
*
https://huggingface.co/lxq007/DiffBIR/resolve/main/general_swinir_v1.ckpt
\
或者从SCNet下载
[
general_swinir_v1
](
http://113.200.138.88:18080/aimodels/DiffBIR/-/raw/main/general_swinir_v1.ckpt?ref_type=heads&inline=false
)
python inference.py \
python inference.py \
--input inputs/demo/general \
--input inputs/demo/general \
...
@@ -152,7 +157,8 @@ Tips:以上dtk驱动、python、torch等DCU相关工具版本需要严格一
...
@@ -152,7 +157,8 @@ Tips:以上dtk驱动、python、torch等DCU相关工具版本需要严格一
模型下载地址:
模型下载地址:
*
https://huggingface.co/lxq007/DiffBIR/resolve/main/face_full_v1.ckpt
*
https://huggingface.co/lxq007/DiffBIR/resolve/main/face_full_v1.ckpt
\
或者从SCNet下载
[
face_full_v1
](
http://113.200.138.88:18080/aimodels/DiffBIR/-/raw/main/face_full_v1.ckpt?ref_type=heads&inline=false
)
for aligned face inputs
for aligned face inputs
...
...
icon.png
0 → 100644
View file @
07447674
59.3 KB
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment