(
self,
size=None,
depth_key="depth",
out_channels=3,
return_valid_mask=True,
exclude_keys_for_resize=None,
keep_raw_depth=False,
)
| 114 | |
| 115 | class DatasetPreprocessor: |
| 116 | def __init__( |
| 117 | self, |
| 118 | size=None, |
| 119 | depth_key="depth", |
| 120 | out_channels=3, |
| 121 | return_valid_mask=True, |
| 122 | exclude_keys_for_resize=None, |
| 123 | keep_raw_depth=False, |
| 124 | ): |
| 125 | self.size = tuple(size) if size is not None else None |
| 126 | self.out_channels = out_channels |
| 127 | self.return_valid_mask = return_valid_mask |
| 128 | self.exclude_keys_for_resize = exclude_keys_for_resize or [] |
| 129 | self.saved_processed_sample = None |
| 130 | self.keep_raw_depth = keep_raw_depth |
| 131 | self.depth_key = depth_key |
| 132 | |
| 133 | def preprocess_sample(self, sample): |
| 134 | """ get dataset name """ |
nothing calls this directly
no outgoing calls
no test coverage detected