Merge pull request #453 from mio2333/main

update parser format to match a global pattern
This commit is contained in:
Kohya S
2023-04-26 21:32:31 +09:00
committed by GitHub

View File

@@ -224,7 +224,7 @@ def main(args):
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 @@ if __name__ == "__main__":
)
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()
# スペルミスしていたオプションを復元する