OSSForks/faceswap
1
0
Fork 0
mirror of https://github.com/zebrajr/faceswap.git synced 2025-12-06 00:20:09 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Extraction: Added an option to skip outputting face images (#1021)

Browse Source
This commit is contained in:
bryanlyon 2020-05-13 04:28:30 -07:00 committed by GitHub
parent 815c843f63
commit 92bc9af957
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
lib/cli/args.py
View File

@ -538,6 +538,13 @@ class ExtractArgs(ExtractConvertArgs):
default=False,
group="settings",
help="Skip frames that already have detected faces in the alignments file"))
argument_list.append(dict(
opts=("-ssf", "--skip-saving-faces"),
action="store_true",
dest="skip_saving_faces",
default=False,
group="settings",
help="Skip saving out the face images"))
return argument_list

3
scripts/extract.py
View File

@ -213,7 +213,8 @@ class Extract(): # pylint:disable=too-few-public-methods
self._check_thread_error()
if is_final:
self._output_processing(extract_media, size)
self._output_faces(saver, extract_media)
if not self._args.skip_saving_faces:
self._output_faces(saver, extract_media)
if self._save_interval and (idx + 1) % self._save_interval == 0:
self._alignments.save()
else: