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

Fix tests for new cli switches

Browse Source
This commit is contained in:
torzdf
2024-04-06 15:48:54 +01:00
parent 95b4431c57
commit b1caa03e3f
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/simple_tests.py
View File

@@ -94,7 +94,7 @@ def train_args(model, model_path, faces, iterations=1, batchsize=2, extra_args="
""" Train command """
py_exe = sys.executable
args = (f"{py_exe} faceswap.py train -A {faces} -B {faces} -m {model_path} -t {model} "
f"-bs {batchsize} -it {iterations} {extra_args}")
f"-b {batchsize} -it {iterations} {extra_args}")
return args.split()
@@ -191,7 +191,7 @@ def main():
(
py_exe, "tools.py", "alignments", "-j", "rename",
"-a", pathjoin(vid_base, "test_alignments.fsa"),
"-fc", pathjoin(vid_base, "faces_sorted"),
"-c", pathjoin(vid_base, "faces_sorted"),
)
)
set_train_config(True)
@@ -202,7 +202,7 @@ def main():
pathjoin(vid_base, "faces"),
iterations=1,
batchsize=1,
extra_args="-wl"))
extra_args="-M"))
set_train_config(False)
was_trained = run_test(