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 dev_segmented_extract,

Browse source 
extracts marked images in 'labelme' tool, that can be used in FANseg training
This commit is contained in:
Colombo 2020-03-08 10:13:15 +04:00
parent 143792fd31
commit 6f4ea69d4d
2 changed files with 69 additions and 51 deletions
Download patch file Download diff file Expand all files Collapse all files

11
main.py
View file

@ -104,6 +104,17 @@ if __name__ == "__main__":
p = subparsers.add_parser( "dev_test", help="")
p.add_argument('--input-dir', required=True, action=fixPathAction, dest="input_dir")
p.set_defaults (func=process_dev_test)
def process_dev_segmented_extract(arguments):
osex.set_process_lowest_prio()
from mainscripts import dev_misc
dev_misc.dev_segmented_extract(arguments.input_dir, arguments.output_dir)
p = subparsers.add_parser( "dev_segmented_extract", help="")
p.add_argument('--input-dir', required=True, action=fixPathAction, dest="input_dir")
p.add_argument('--output-dir', required=True, action=fixPathAction, dest="output_dir")
p.set_defaults (func=process_dev_segmented_extract)
def process_sort(arguments):
osex.set_process_lowest_prio()