Add sapiens2 matting as a mask generator. Begin transition to model paths and model folders.

This commit is contained in:
Jaret Burkett
2026-05-20 08:56:16 -06:00
parent 15d4fb89ff
commit aecd554128
8 changed files with 209 additions and 34 deletions

View File

@@ -149,6 +149,7 @@ class BaseCaptioner(BaseExtensionProcess):
def find_files(self):
# recursivly find all the files in the path_to_caption with the specified extensions and save the paths to self.file_paths
for root, dirs, files in os.walk(self.caption_config.path_to_caption):
dirs[:] = [d for d in dirs if d != "_controls"]
for file in files:
if any(
file.lower().endswith(f".{ext}") and not file.startswith(".")