From 57bc2abf41756f9d2d14b1049af3f198312820a1 Mon Sep 17 00:00:00 2001 From: mio Date: Wed, 26 Apr 2023 20:11:32 +0800 Subject: [PATCH] update parser format to match a global pattern --- finetune/tag_images_by_wd14_tagger.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/finetune/tag_images_by_wd14_tagger.py b/finetune/tag_images_by_wd14_tagger.py index 80a55716f..91e4f573e 100644 --- a/finetune/tag_images_by_wd14_tagger.py +++ b/finetune/tag_images_by_wd14_tagger.py @@ -224,7 +224,7 @@ def run_batch(path_imgs): print("done!") -if __name__ == "__main__": +def setup_parser() -> argparse.ArgumentParser: parser = argparse.ArgumentParser() parser.add_argument("train_data_dir", type=str, help="directory for train images / 学習画像データのディレクトリ") parser.add_argument( @@ -284,6 +284,11 @@ def run_batch(path_imgs): ) parser.add_argument("--frequency_tags", action="store_true", help="Show frequency of tags for images / 画像ごとのタグの出現頻度を表示する") + return parser + +if __name__ == "__main__": + parser = setup_parser() + args = parser.parse_args() # スペルミスしていたオプションを復元する