github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 13:02:15 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added options --alpha and --transfer color

Browse source 
--alpha will make export png with alpha channel (export only face)
--transfercolor will transfer color from original DST image to fake
This commit is contained in:
MaksV79 2018-11-17 18:12:38 +03:00 • committed by GitHub
parent 46eea1c113
commit 3dcedf6bf3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

24
main.py
View file

@ -1,4 +1,4 @@
import os import os
import sys import sys
import argparse import argparse
from utils import Path_utils from utils import Path_utils
@ -136,7 +136,19 @@ if __name__ == "__main__":
arguments.blur_mask_modifier = int ( input ("Choose blur mask modifier [-100..200] (default 0) : ") ) arguments.blur_mask_modifier = int ( input ("Choose blur mask modifier [-100..200] (default 0) : ") )
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:
arguments.alpha = bool ( {"1":True,"0":False}[input("Export png with alpha channel? [0..1] (default 0) : ").lower()] )
except:
arguments.alpha = False
if (arguments.transfercolor == False): #if argument was not passed in command line, ask it
try:
arguments.transfercolor = bool ( {"1":True,"0":False}[input("Transfer color from original DST image? [0..1] (default 0) : ").lower()] )
except:
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)
@ -152,7 +164,9 @@ if __name__ == "__main__":
masked_hist_match = arguments.masked_hist_match, masked_hist_match = arguments.masked_hist_match,
erode_mask_modifier = arguments.erode_mask_modifier, erode_mask_modifier = arguments.erode_mask_modifier,
blur_mask_modifier = arguments.blur_mask_modifier, blur_mask_modifier = arguments.blur_mask_modifier,
force_best_gpu_idx = arguments.force_best_gpu_idx force_best_gpu_idx = arguments.force_best_gpu_idx,
alpha = arguments.alpha,
transfercolor = arguments.transfercolor,
) )
convert_parser = subparsers.add_parser( "convert", help="Converter") convert_parser = subparsers.add_parser( "convert", help="Converter")
@ -167,6 +181,8 @@ if __name__ == "__main__":
convert_parser.add_argument('--erode-mask-modifier', type=int, dest="erode_mask_modifier", default=0, help="Automatic erode mask modifier. Valid range [-100..100].") convert_parser.add_argument('--erode-mask-modifier', type=int, dest="erode_mask_modifier", default=0, help="Automatic erode mask modifier. Valid range [-100..100].")
convert_parser.add_argument('--blur-mask-modifier', type=int, dest="blur_mask_modifier", default=0, help="Automatic blur mask modifier. Valid range [-100..200].") convert_parser.add_argument('--blur-mask-modifier', type=int, dest="blur_mask_modifier", default=0, help="Automatic blur mask modifier. Valid range [-100..200].")
convert_parser.add_argument('--debug', action="store_true", dest="debug", default=False, help="Debug converter.") convert_parser.add_argument('--debug', action="store_true", dest="debug", default=False, help="Debug converter.")
convert_parser.add_argument('--alpha', action="store_true", dest="alpha", default=False, help="alpha channel.")
convert_parser.add_argument('--transfercolor', action="store_true", dest="transfercolor", default=False, help="transfer color from dst to merged.")
convert_parser.add_argument('--force-best-gpu-idx', type=int, dest="force_best_gpu_idx", default=-1, help="Force to choose this GPU idx as best.") convert_parser.add_argument('--force-best-gpu-idx', type=int, dest="force_best_gpu_idx", default=-1, help="Force to choose this GPU idx as best.")
convert_parser.set_defaults(func=process_convert) convert_parser.set_defaults(func=process_convert)
@ -185,4 +201,4 @@ if __name__ == "__main__":
''' '''
import code import code
code.interact(local=dict(globals(), **locals())) code.interact(local=dict(globals(), **locals()))
''' '''