Args: data_dir: path to the dataset root directory split: one of "train", "val", "test"
(
self,
data_dir: str,
split: str = "train",
# Add other params from config
)
| 36 | """ |
| 37 | |
| 38 | def __init__( |
| 39 | self, |
| 40 | data_dir: str, |
| 41 | split: str = "train", |
| 42 | # Add other params from config |
| 43 | ): |
| 44 | """ |
| 45 | Args: |
| 46 | data_dir: path to the dataset root directory |
| 47 | split: one of "train", "val", "test" |
| 48 | """ |
| 49 | self.data_dir = Path(data_dir) |
| 50 | self.split = split |
| 51 | |
| 52 | # TODO: Load file list / metadata |
| 53 | # self.samples = self._load_samples() |
| 54 | |
| 55 | def _load_samples(self): |
| 56 | """Load sample paths/metadata for the given split. |
nothing calls this directly
no outgoing calls
no test coverage detected