github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-05 20:42:11 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

added 5.XSeg) data_dst/data_src mask for XSeg trainer - remove.bat

Browse source 
removes labeled xseg polygons from the extracted frames
This commit is contained in:
Colombo 2020-04-06 21:45:46 +04:00
parent b6b3936bcd
commit 5b4d023712
3 changed files with 69 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

12
main.py
View file

@ -278,8 +278,7 @@ if __name__ == "__main__":
p.set_defaults (func=process_xsegapply)
p = xseg_parser.add_parser( "remove", help="Remove XSeg from the extracted faces.")
p = xseg_parser.add_parser( "remove", help="Remove applied XSeg masks from the extracted faces.")
def process_xsegremove(arguments):
osex.set_process_lowest_prio()
from mainscripts import XSegUtil
@ -288,6 +287,15 @@ if __name__ == "__main__":
p.set_defaults (func=process_xsegremove)
p = xseg_parser.add_parser( "remove_labels", help="Remove XSeg labels from the extracted faces.")
def process_xsegremovelabels(arguments):
osex.set_process_lowest_prio()
from mainscripts import XSegUtil
XSegUtil.remove_xseg_labels (Path(arguments.input_dir) )
p.add_argument('--input-dir', required=True, action=fixPathAction, dest="input_dir")
p.set_defaults (func=process_xsegremovelabels)
p = xseg_parser.add_parser( "fetch", help="Copies faces containing XSeg polygons in <input_dir>_xseg dir.")
def process_xsegfetch(arguments):