Add ability to pull control images from same folder group

This commit is contained in:
Jaret Burkett committed 2026-07-08 05:21:31 -06:00
1 parent 7bceec3b07
commit dd08579eda
2 files changed
+29 -3

No files matched your search

+4
View File
@@ -923,6 +923,10 @@ class DatasetConfig:
self.flip_y: bool = kwargs.get('flip_y', False)
self.augments: List[str] = kwargs.get('augments', [])
self.control_path: Union[str,List[str]] = kwargs.get('control_path', None) # depth maps, etc
# pull a random control image from the same folder as the image. Useful for folder grouped pairs.
self.control_from_same_folder: bool = kwargs.get('control_from_same_folder', False)
self.num_controls_from_same_folder: int = kwargs.get('num_controls_from_same_folder', 1)
if self.control_path == '':
self.control_path = None