Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Overbill1683
Stable Diffusion Webui
提交
3fee3c34
S
Stable Diffusion Webui
项目概览
Overbill1683
/
Stable Diffusion Webui
11 个月 前同步成功
通知
1758
Star
81
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
分析
仓库
DevOps
项目成员
Pages
S
Stable Diffusion Webui
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Pages
分析
分析
仓库分析
DevOps
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
提交
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
3fee3c34
编写于
7月 12, 2023
作者:
M
missionfloyd
提交者:
GitHub
7月 12, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Save img2img batch with images.save_image()
上级
f865d3e1
变更
1
显示空白变更内容
内联
并排
Showing
1 changed file
with
5 addition
and
4 deletion
+5
-4
modules/img2img.py
modules/img2img.py
+5
-4
未找到文件。
modules/img2img.py
浏览文件 @
3fee3c34
...
@@ -8,6 +8,7 @@ from modules import sd_samplers
...
@@ -8,6 +8,7 @@ from modules import sd_samplers
from
modules.generation_parameters_copypaste
import
create_override_settings_dict
from
modules.generation_parameters_copypaste
import
create_override_settings_dict
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
...
@@ -84,17 +85,17 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args, to_scale=Fal
...
@@ -84,17 +85,17 @@ 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
)
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
(
output_dir
,
exist_ok
=
True
)
os
.
makedirs
(
output_dir
,
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
,
filename
)
)
save_image
(
processed_image
,
output_dir
,
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
,
*
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
,
*
args
):
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录