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:11:55 -08:00 committed by GitHub
parent 40aad268ac
commit a52c3b8fb3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
finetune/tag_images_by_wd14_tagger.py
View File

@ -125,7 +125,7 @@ def main(args):
tag_text = "" tag_text = ""
for i, p in enumerate(prob[4:]): # numpyとか使うのが良いけど、まあそれほど数も多くないのでループで for i, p in enumerate(prob[4:]): # numpyとか使うのが良いけど、まあそれほど数も多くないのでループで
if p >= args.thresh and i < len(tags): if p >= args.thresh and i < len(tags):
tag_text += ", " + tags[i] tag_text += ", " + (tags[i].replace("_", " ") if args.replace_underscores else tags[i])
if len(tag_text) > 0: if len(tag_text) > 0:
tag_text = tag_text[2:] # 最初の ", " を消す tag_text = tag_text[2:] # 最初の ", " を消す
@ -190,6 +190,7 @@ if __name__ == '__main__':
help="extension of caption file (for backward compatibility) / 出力されるキャプションファイルの拡張子（スペルミスしていたのを残してあります）") help="extension of caption file (for backward compatibility) / 出力されるキャプションファイルの拡張子（スペルミスしていたのを残してあります）")
parser.add_argument("--caption_extension", type=str, default=".txt", help="extension of caption file / 出力されるキャプションファイルの拡張子") parser.add_argument("--caption_extension", type=str, default=".txt", help="extension of caption file / 出力されるキャプションファイルの拡張子")
parser.add_argument("--debug", action="store_true", help="debug mode") parser.add_argument("--debug", action="store_true", help="debug mode")
parser.add_argument("--replace_underscores", action="store_true", help="replace underscores in tags with spaces / タグのアンダースコアをスペースに置き換える")
args = parser.parse_args() args = parser.parse_args()