From a533ec4310196a084e1412b04b2fdcf3637474de Mon Sep 17 00:00:00 2001 From: jh Date: Thu, 11 Mar 2021 23:30:02 -0800 Subject: [PATCH] fix: model path for preview --- mainscripts/Trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index 4056fe1..0e81831 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -403,7 +403,7 @@ def main(**kwargs): batch_size, zoom) # io.show_image(wnd_name, preview_pane_image) - model_path = Path(args.get('model_path', '')) + model_path = Path(kwargs.get('model_path', '')) filename = 'preview.jpg' preview_file = str(model_path / filename) cv2.imwrite(preview_file, preview_pane_image)