(self, x)
| 171 | self._d2_viz_color_mode = D2_COLORMODE_MAPPING[cfg.VIS.D2.DATALOADER.COLOR_MODE] |
| 172 | |
| 173 | def visualize(self, x): |
| 174 | # Assumption: dataloader produce CHW images. |
| 175 | img = d2_utils.convert_image_to_rgb(x['image'].permute(1, 2, 0), self._input_format) |
| 176 | |
| 177 | viz_images = OrderedDict() |
| 178 | |
| 179 | # d2 instance viz (2D box, mask, keypoints) |
| 180 | gt_instances = x['instances'] |
| 181 | viz_image = draw_gt_instances_d2(gt_instances, img, self._metadata, self._scale, self._d2_viz_color_mode) |
| 182 | viz_images['viz_gt_instances_d2'] = viz_image |
| 183 | |
| 184 | return viz_images |
nothing calls this directly
no test coverage detected