From d0128d18be009c5e221db96d53b6045bdd5af04f Mon Sep 17 00:00:00 2001 From: rockerBOO Date: Fri, 14 Feb 2025 14:14:57 -0500 Subject: [PATCH] Add resize interpolation CLI option --- library/train_util.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/library/train_util.py b/library/train_util.py index a07834ad..d41d1ff3 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -4530,7 +4530,13 @@ def add_dataset_arguments( action="store_true", help="make bucket for each image without upscaling / 画像を拡大せずbucketを作成します", ) - + parser.add_argument( + "--resize_interpolation", + type=str, + default=None, + choices=["lanczos", "nearest", "bilinear", "linear", "bicubic", "cubic", "area"], + help="Resize interpolation when required. Default: area Options: lanczos, nearest, bilinear, bicubic, area / 必要に応じてサイズ補間を変更します。デフォルト: area オプション: lanczos, nearest, bilinear, bicubic, area", + ) parser.add_argument( "--token_warmup_min", type=int,