(self, map_mode)
| 109 | |
| 110 | @params("identity", "xflip", "yflip", "xyflip", "random") |
| 111 | def test_remap(self, map_mode): |
| 112 | maps = [update_map(mode=map_mode, shape=self.img_size, nimages=self.batch_size)] |
| 113 | dpipe = remap_pipe("dali", maps, self.img_size, **self.common_dali_pipe_params) |
| 114 | cpipe = remap_pipe( |
| 115 | "cv", |
| 116 | maps, |
| 117 | self.img_size, |
| 118 | exec_async=False, |
| 119 | exec_pipelined=False, |
| 120 | **self.common_dali_pipe_params, |
| 121 | ) |
| 122 | self._compare_pipelines_pixelwise(dpipe, cpipe, N_iterations=2, eps=0.01) |
| 123 | |
| 124 | def benchmark_remap_against_cv(self, map_mode): |
| 125 | import torch.cuda.nvtx as nvtx |
nothing calls this directly
no test coverage detected