StableDiffusion/KohyaSS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add replace underscores option

Browse Source
This commit is contained in:
saltacc 2023-03-05 13:10:39 -08:00 committed by GitHub
parent 7be7f6d2c4
commit 40aad268ac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
library/wd14_caption_gui.py
View File

@ -5,7 +5,7 @@ from .common_gui import get_folder_path
import os import os
def caption_images(train_data_dir, caption_extension, batch_size, thresh): def caption_images(train_data_dir, caption_extension, batch_size, thresh, replace_underscores):
# Check for caption_text_input # Check for caption_text_input
# if caption_text_input == "": # if caption_text_input == "":
# msgbox("Caption text is missing...") # msgbox("Caption text is missing...")
@ -24,6 +24,7 @@ def caption_images(train_data_dir, caption_extension, batch_size, thresh):
run_cmd = f'accelerate launch "./finetune/tag_images_by_wd14_tagger.py"' run_cmd = f'accelerate launch "./finetune/tag_images_by_wd14_tagger.py"'
run_cmd += f' --batch_size="{int(batch_size)}"' run_cmd += f' --batch_size="{int(batch_size)}"'
run_cmd += f' --thresh="{thresh}"' run_cmd += f' --thresh="{thresh}"'
run_cmd += f' --replace_underscores' if replace_underscores else ''
if caption_extension != '': if caption_extension != '':
run_cmd += f' --caption_extension="{caption_extension}"' run_cmd += f' --caption_extension="{caption_extension}"'
run_cmd += f' "{train_data_dir}"' run_cmd += f' "{train_data_dir}"'
@ -75,11 +76,17 @@ def gradio_wd14_caption_gui_tab():
batch_size = gr.Number( batch_size = gr.Number(
value=1, label='Batch size', interactive=True value=1, label='Batch size', interactive=True
) )
replace_underscores = gr.Checkbox(
label='Replace underscores in filenames with spaces',
value=False,
interactive=True,
)
caption_button = gr.Button('Caption images') caption_button = gr.Button('Caption images')
caption_button.click( caption_button.click(
caption_images, caption_images,
inputs=[train_data_dir, caption_extension, batch_size, thresh], inputs=[train_data_dir, caption_extension, batch_size, thresh, replace_underscores],
show_progress=False, show_progress=False,
) )