Skip to content

Commit

Permalink
Bugfix: mask tool. Prevent hangiing when input is folder of frames
Browse files Browse the repository at this point in the history
  • Loading branch information
torzdf committed Mar 13, 2024
1 parent 23944c3 commit af95dbb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tools/mask/mask.py
Original file line number Diff line number Diff line change
Expand Up @@ -149,6 +149,7 @@ def __init__(self, arguments: Namespace) -> None:

self._loader = Loader(arguments.input, self._input_is_faces)
self._alignments = self._get_alignments(arguments.alignments, arguments.input)
self._loader.add_alignments(self._alignments)

self._output = Output(arguments, self._alignments, self._loader.file_list)

Expand Down Expand Up @@ -231,7 +232,6 @@ def _get_alignments(self, alignments: str | None, input_location: str) -> Alignm
filename = "alignments"

retval = Alignments(folder, filename=filename)
self._loader.add_alignments(retval)
return retval

def _save_output(self, media: ExtractMedia) -> None:
Expand Down

0 comments on commit af95dbb

Please sign in to comment.