| 47 | |
| 48 | |
| 49 | class _TestCompose(Compose): |
| 50 | |
| 51 | def __call__(self, data, meta, lazy): |
| 52 | data = self.transforms[0](data) # ensure channel first |
| 53 | data = self.transforms[1](data, lazy=lazy) # spacing |
| 54 | meta = data.meta |
| 55 | if len(self.transforms) == 3: |
| 56 | return self.transforms[2](data), meta # image contrast |
| 57 | return data, meta |
| 58 | |
| 59 | |
| 60 | class TestImageDataset(unittest.TestCase): |
no outgoing calls
searching dependent graphs…