mirror/DeepCreamPy
1
0
Fork 0
mirror of https://github.com/Deepshift/DeepCreamPy.git synced 2024-12-01 04:00:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added --autoclose argument

Browse Source
This commit is contained in:
qT2DyV 2019-07-21 21:06:08 +01:00
parent 8996c4204c
commit 238b02185e
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
config.py
View File

@ -29,6 +29,9 @@ 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('--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') 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() args = parser.parse_args()
return args return args

1
decensor.py
View File

@ -95,6 +95,7 @@ class Decensor:
print("--------------------------------------------------------------------------") print("--------------------------------------------------------------------------")
if(self.files_removed is not None): if(self.files_removed is not None):
file.error_messages(None, self.files_removed) file.error_messages(None, self.files_removed)
if(self.args.autoclose == False):
input("\nPress anything to end...") input("\nPress anything to end...")
#decensors one image at a time #decensors one image at a time