diff --git a/config.py b/config.py index 0c27bbc..84a00df 100644 --- a/config.py +++ b/config.py @@ -29,8 +29,11 @@ def get_args(): parser.add_argument('--mask_color', dest='mask_color', default=[0,255,0], type=str2floatarr, help='rgb color of the mask, comma seperated.') parser.add_argument('--is_mosaic', dest='is_mosaic', default='False', type=str2bool, help='true if image has mosaic censoring, false otherwise') + #Misc settings + parser.add_argument('--autoclose', dest='autoclose', default='False', type=str2bool, help='true will close the program when it finishes') + args = parser.parse_args() return args if __name__ == '__main__': - get_args() \ No newline at end of file + get_args() diff --git a/decensor.py b/decensor.py index ac00885..675d1dc 100644 --- a/decensor.py +++ b/decensor.py @@ -96,10 +96,11 @@ class Decensor: print("--------------------------------------------------------------------------") if(self.files_removed is not None): file.error_messages(None, self.files_removed) - input("\nPress anything to end...") - - # decensors one image at a time - # TODO: decensor all cropped parts of the same image in a batch (then i need input for colored an array of those images and make additional changes) + if(self.args.autoclose == False): + input("\nPress anything to end...") + + #decensors one image at a time + #TODO: decensor all cropped parts of the same image in a batch (then i need input for colored an array of those images and make additional changes) def decensor_image(self, ori, colored, file_name=None): width, height = ori.size # save the alpha channel if the image has an alpha channel