Add replace underscores option

finetune/tag_images_by_wd14_tagger.py

@@ -125,7 +125,7 @@ def main(args):
       tag_text = ""
       for i, p in enumerate(prob[4:]):                # numpyとか使うのが良いけど、まあそれほど数も多くないのでループで
         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:
         tag_text = tag_text[2:]                   # 最初の ", " を消す
@@ -190,6 +190,7 @@ if __name__ == '__main__':
                       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("--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()