Args: image (ndarray): (H,W) or (H,W,C) ndarray of type uint8 in range [0, 255], or floating point in range [0, 1] or [0, 255]. The meaning of C is up to users. boxes (ndarray or None): Nx4 float32 boxes in XYXY_ABS mode se
(
self,
image: np.ndarray,
*,
boxes: Optional[np.ndarray] = None,
sem_seg: Optional[np.ndarray] = None,
)
| 640 | |
| 641 | # TODO maybe should support more builtin data types here |
| 642 | def __init__( |
| 643 | self, |
| 644 | image: np.ndarray, |
| 645 | *, |
| 646 | boxes: Optional[np.ndarray] = None, |
| 647 | sem_seg: Optional[np.ndarray] = None, |
| 648 | ): |
| 649 | """ |
| 650 | Args: |
| 651 | image (ndarray): (H,W) or (H,W,C) ndarray of type uint8 in range [0, 255], or |
| 652 | floating point in range [0, 1] or [0, 255]. The meaning of C is up |
| 653 | to users. |
| 654 | boxes (ndarray or None): Nx4 float32 boxes in XYXY_ABS mode |
| 655 | sem_seg (ndarray or None): HxW uint8 semantic segmentation mask. Each element |
| 656 | is an integer label of pixel. |
| 657 | """ |
| 658 | _check_img_dtype(image) |
| 659 | self.image = image |
| 660 | self.boxes = boxes |
| 661 | self.sem_seg = sem_seg |
| 662 | |
| 663 | def transform(self, tfm: Transform) -> None: |
| 664 | """ |
no test coverage detected