diff options
| author | Milly <milly.ca@gmail.com> | 2022-10-20 13:56:45 +0000 |
|---|---|---|
| committer | Milly <milly.ca@gmail.com> | 2022-10-20 14:35:01 +0000 |
| commit | 85dd62c4c7635b8e21a75f140d093036069e97a1 (patch) | |
| tree | d69c5796f6af6053092642bd6c367530f9c5195d /modules/ui.py | |
| parent | 9681419e422515e42444e0174355b760645a846f (diff) | |
| download | stable-diffusion-webui-gfx803-85dd62c4c7635b8e21a75f140d093036069e97a1.tar.gz stable-diffusion-webui-gfx803-85dd62c4c7635b8e21a75f140d093036069e97a1.tar.bz2 stable-diffusion-webui-gfx803-85dd62c4c7635b8e21a75f140d093036069e97a1.zip | |
train: ui: added `Split image threshold` and `Split image overlap ratio` to preprocess
Diffstat (limited to 'modules/ui.py')
| -rw-r--r-- | modules/ui.py | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/modules/ui.py b/modules/ui.py index a2dbd41e..bc7f3330 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -1240,10 +1240,14 @@ def create_ui(wrap_gradio_gpu_call): with gr.Row():
process_flip = gr.Checkbox(label='Create flipped copies')
- process_split = gr.Checkbox(label='Split oversized images into two')
+ process_split = gr.Checkbox(label='Split oversized images')
process_caption = gr.Checkbox(label='Use BLIP for caption')
process_caption_deepbooru = gr.Checkbox(label='Use deepbooru for caption', visible=True if cmd_opts.deepdanbooru else False)
+ with gr.Row(visible=False) as process_split_extra_row:
+ process_split_threshold = gr.Slider(label='Split image threshold', value=0.5, minimum=0.0, maximum=1.0, step=0.05)
+ process_overlap_ratio = gr.Slider(label='Split image overlap ratio', value=0.2, minimum=0.0, maximum=0.9, step=0.05)
+
with gr.Row():
with gr.Column(scale=3):
gr.HTML(value="")
@@ -1251,6 +1255,12 @@ def create_ui(wrap_gradio_gpu_call): with gr.Column():
run_preprocess = gr.Button(value="Preprocess", variant='primary')
+ process_split.change(
+ fn=lambda show: gr_show(show),
+ inputs=[process_split],
+ outputs=[process_split_extra_row],
+ )
+
with gr.Tab(label="Train"):
gr.HTML(value="<p style='margin-bottom: 0.7em'>Train an embedding; must specify a directory with a set of 1:1 ratio images</p>")
with gr.Row():
@@ -1327,7 +1337,9 @@ def create_ui(wrap_gradio_gpu_call): process_flip,
process_split,
process_caption,
- process_caption_deepbooru
+ process_caption_deepbooru,
+ process_split_threshold,
+ process_overlap_ratio,
],
outputs=[
ti_output,
|