Convert different format of masks or polygons to a tuple of masks and polygons. Returns: list[GenericMask]:
(self, masks_or_polygons)
| 1225 | return np.asarray(boxes) |
| 1226 | |
| 1227 | def _convert_masks(self, masks_or_polygons): |
| 1228 | """ |
| 1229 | Convert different format of masks or polygons to a tuple of masks and polygons. |
| 1230 | |
| 1231 | Returns: |
| 1232 | list[GenericMask]: |
| 1233 | """ |
| 1234 | |
| 1235 | m = masks_or_polygons |
| 1236 | if isinstance(m, PolygonMasks): |
| 1237 | m = m.polygons |
| 1238 | if isinstance(m, BitMasks): |
| 1239 | m = m.tensor.numpy() |
| 1240 | if isinstance(m, torch.Tensor): |
| 1241 | m = m.numpy() |
| 1242 | ret = [] |
| 1243 | for x in m: |
| 1244 | if isinstance(x, GenericMask): |
| 1245 | ret.append(x) |
| 1246 | else: |
| 1247 | ret.append(GenericMask(x, self.output.height, self.output.width)) |
| 1248 | return ret |
| 1249 | |
| 1250 | def _draw_text_in_mask(self, binary_mask, text, color): |
| 1251 | """ |
no test coverage detected