cascade.utils.vision#
- class cascade.utils.vision.FolderImageDataset(root: str, backend: Literal['cv2', 'PIL'] = 'PIL', *args: Any, **kwargs: Any)[source]#
Simple dataset for image folder with lazy loading. Accepts the path to the folder with images. In each __getitem__ call invokes opencv imread on image and returns it if it exists.
Supports opencv or pillow backends