Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Overbill1683
Stable Diffusion Webui
提交
ed0512c7
S
Stable Diffusion Webui
项目概览
Overbill1683
/
Stable Diffusion Webui
10 个月 前同步成功
通知
1748
Star
81
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
分析
仓库
DevOps
项目成员
Pages
S
Stable Diffusion Webui
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Pages
分析
分析
仓库分析
DevOps
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
提交
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
ed0512c7
编写于
7月 13, 2023
作者:
A
AUTOMATIC1111
提交者:
GitHub
7月 13, 2023
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #11747 from AUTOMATIC1111/img2img-save
Save img2img batch with images.save_image()
上级
cc0a3cc4
e0218c4f
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
5 addition
and
4 deletion
+5
-4
modules/img2img.py
modules/img2img.py
+5
-4
未找到文件。
modules/img2img.py
浏览文件 @
ed0512c7
...
@@ -10,6 +10,7 @@ from modules import sd_samplers, images as imgutil
...
@@ -10,6 +10,7 @@ from modules import sd_samplers, images as imgutil
from
modules.generation_parameters_copypaste
import
create_override_settings_dict
,
parse_generation_parameters
from
modules.generation_parameters_copypaste
import
create_override_settings_dict
,
parse_generation_parameters
from
modules.processing
import
Processed
,
StableDiffusionProcessingImg2Img
,
process_images
from
modules.processing
import
Processed
,
StableDiffusionProcessingImg2Img
,
process_images
from
modules.shared
import
opts
,
state
from
modules.shared
import
opts
,
state
from
modules.images
import
save_image
import
modules.shared
as
shared
import
modules.shared
as
shared
import
modules.processing
as
processing
import
modules.processing
as
processing
from
modules.ui
import
plaintext_to_html
from
modules.ui
import
plaintext_to_html
...
@@ -113,18 +114,18 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args, to_scale=Fal
...
@@ -113,18 +114,18 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args, to_scale=Fal
proc
=
process_images
(
p
)
proc
=
process_images
(
p
)
for
n
,
processed_image
in
enumerate
(
proc
.
images
):
for
n
,
processed_image
in
enumerate
(
proc
.
images
):
filename
=
image_path
.
name
filename
=
image_path
.
stem
infotext
=
proc
.
infotext
(
p
,
n
)
relpath
=
os
.
path
.
dirname
(
os
.
path
.
relpath
(
image
,
input_dir
))
relpath
=
os
.
path
.
dirname
(
os
.
path
.
relpath
(
image
,
input_dir
))
if
n
>
0
:
if
n
>
0
:
left
,
right
=
os
.
path
.
splitext
(
filename
)
filename
+=
f
"-
{
n
}
"
filename
=
f
"
{
left
}
-
{
n
}{
right
}
"
if
not
save_normally
:
if
not
save_normally
:
os
.
makedirs
(
os
.
path
.
join
(
output_dir
,
relpath
),
exist_ok
=
True
)
os
.
makedirs
(
os
.
path
.
join
(
output_dir
,
relpath
),
exist_ok
=
True
)
if
processed_image
.
mode
==
'RGBA'
:
if
processed_image
.
mode
==
'RGBA'
:
processed_image
=
processed_image
.
convert
(
"RGB"
)
processed_image
=
processed_image
.
convert
(
"RGB"
)
processed_image
.
save
(
os
.
path
.
join
(
output_dir
,
relpath
,
filename
)
)
save_image
(
processed_image
,
os
.
path
.
join
(
output_dir
,
relpath
),
None
,
extension
=
opts
.
samples_format
,
info
=
infotext
,
forced_filename
=
filename
,
save_to_dirs
=
False
)
def
img2img
(
id_task
:
str
,
mode
:
int
,
prompt
:
str
,
negative_prompt
:
str
,
prompt_styles
,
init_img
,
sketch
,
init_img_with_mask
,
inpaint_color_sketch
,
inpaint_color_sketch_orig
,
init_img_inpaint
,
init_mask_inpaint
,
steps
:
int
,
sampler_index
:
int
,
mask_blur
:
int
,
mask_alpha
:
float
,
inpainting_fill
:
int
,
restore_faces
:
bool
,
tiling
:
bool
,
n_iter
:
int
,
batch_size
:
int
,
cfg_scale
:
float
,
image_cfg_scale
:
float
,
denoising_strength
:
float
,
seed
:
int
,
subseed
:
int
,
subseed_strength
:
float
,
seed_resize_from_h
:
int
,
seed_resize_from_w
:
int
,
seed_enable_extras
:
bool
,
selected_scale_tab
:
int
,
height
:
int
,
width
:
int
,
scale_by
:
float
,
resize_mode
:
int
,
inpaint_full_res
:
bool
,
inpaint_full_res_padding
:
int
,
inpainting_mask_invert
:
int
,
img2img_batch_input_dir
:
str
,
img2img_batch_output_dir
:
str
,
img2img_batch_inpaint_mask_dir
:
str
,
override_settings_texts
,
img2img_batch_use_png_info
:
bool
,
img2img_batch_png_info_props
:
list
,
img2img_batch_png_info_dir
:
str
,
request
:
gr
.
Request
,
*
args
):
def
img2img
(
id_task
:
str
,
mode
:
int
,
prompt
:
str
,
negative_prompt
:
str
,
prompt_styles
,
init_img
,
sketch
,
init_img_with_mask
,
inpaint_color_sketch
,
inpaint_color_sketch_orig
,
init_img_inpaint
,
init_mask_inpaint
,
steps
:
int
,
sampler_index
:
int
,
mask_blur
:
int
,
mask_alpha
:
float
,
inpainting_fill
:
int
,
restore_faces
:
bool
,
tiling
:
bool
,
n_iter
:
int
,
batch_size
:
int
,
cfg_scale
:
float
,
image_cfg_scale
:
float
,
denoising_strength
:
float
,
seed
:
int
,
subseed
:
int
,
subseed_strength
:
float
,
seed_resize_from_h
:
int
,
seed_resize_from_w
:
int
,
seed_enable_extras
:
bool
,
selected_scale_tab
:
int
,
height
:
int
,
width
:
int
,
scale_by
:
float
,
resize_mode
:
int
,
inpaint_full_res
:
bool
,
inpaint_full_res_padding
:
int
,
inpainting_mask_invert
:
int
,
img2img_batch_input_dir
:
str
,
img2img_batch_output_dir
:
str
,
img2img_batch_inpaint_mask_dir
:
str
,
override_settings_texts
,
img2img_batch_use_png_info
:
bool
,
img2img_batch_png_info_props
:
list
,
img2img_batch_png_info_dir
:
str
,
request
:
gr
.
Request
,
*
args
):
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录