(self)
| 63 | v.draw_dataset_dict(dic) |
| 64 | |
| 65 | def test_overlay_instances(self): |
| 66 | img, boxes, labels, polygons, masks = self._random_data() |
| 67 | |
| 68 | v = Visualizer(img, self.metadata) |
| 69 | output = v.overlay_instances(masks=polygons, boxes=boxes, labels=labels).get_image() |
| 70 | self.assertEqual(output.shape, img.shape) |
| 71 | |
| 72 | # Test 2x scaling |
| 73 | v = Visualizer(img, self.metadata, scale=2.0) |
| 74 | output = v.overlay_instances(masks=polygons, boxes=boxes, labels=labels).get_image() |
| 75 | self.assertEqual(output.shape[0], img.shape[0] * 2) |
| 76 | |
| 77 | # Test overlay masks |
| 78 | v = Visualizer(img, self.metadata) |
| 79 | output = v.overlay_instances(masks=masks, boxes=boxes, labels=labels).get_image() |
| 80 | self.assertEqual(output.shape, img.shape) |
| 81 | |
| 82 | def test_overlay_instances_no_boxes(self): |
| 83 | img, boxes, labels, polygons, _ = self._random_data() |
nothing calls this directly
no test coverage detected