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 a720e59..29a3fcf 100644 --- a/decensor.py +++ b/decensor.py @@ -95,7 +95,8 @@ class Decensor: print("--------------------------------------------------------------------------") if(self.files_removed is not None): file.error_messages(None, self.files_removed) - input("\nPress anything to end...") + 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)