diff --git a/scripts/main.py b/scripts/main.py index b07c395..ece489a 100644 --- a/scripts/main.py +++ b/scripts/main.py @@ -64,7 +64,7 @@ def processing(single_image, batch_image, input_dir, output_dir, show_result, in images.save_image( Image.fromarray(mask), path=output_dir, - basename=basename+"_mask", + basename=basename+"-mask", extension=ext, ) if (show_result):