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

5
config.py
View File

@ -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('--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
if __name__ == '__main__': if __name__ == '__main__':
get_args() get_args()

3
decensor.py
View File

@ -95,7 +95,8 @@ 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)
input("\nPress anything to end...") if(self.args.autoclose == False):
input("\nPress anything to end...")
#decensors one image at a time #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) #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)