CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix options --alpha and --transfercolor

Browse Source 
Always use params from dialog
This commit is contained in:
MaksV79
2018-11-19 23:31:06 +03:00
committed by GitHub
parent b4af9d3c36
commit d47ef23df1
1 changed files with 8 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
+8 -10
View File
@@ -137,17 +137,15 @@ if __name__ == "__main__":
except: except:
arguments.blur_mask_modifier = 0 arguments.blur_mask_modifier = 0
if (arguments.alpha == False): #if argument was not passed in command line, ask it try:
try: arguments.alpha = bool ( {"1":True,"0":False}[input("Export png with alpha channel? [0..1] (default 0) : ").lower()] )
arguments.alpha = bool ( {"1":True,"0":False}[input("Export png with alpha channel? [0..1] (default 0) : ").lower()] ) except:
except: arguments.alpha = False
arguments.alpha = False
if (arguments.transfercolor == False): #if argument was not passed in command line, ask it try:
try: arguments.transfercolor = bool ( {"1":True,"0":False}[input("Transfer color from original DST image? [0..1] (default 0) : ").lower()] )
arguments.transfercolor = bool ( {"1":True,"0":False}[input("Transfer color from original DST image? [0..1] (default 0) : ").lower()] ) except:
except: arguments.transfercolor = False
arguments.transfercolor = False
arguments.erode_mask_modifier = np.clip ( int(arguments.erode_mask_modifier), -100, 100) arguments.erode_mask_modifier = np.clip ( int(arguments.erode_mask_modifier), -100, 100)
arguments.blur_mask_modifier = np.clip ( int(arguments.blur_mask_modifier), -100, 200) arguments.blur_mask_modifier = np.clip ( int(arguments.blur_mask_modifier), -100, 200)