cutcutcodec.core.nn.dataset.image

An image dataset.

Classes

ImageDataset(root, shape, *[, dataaug])

A specific dataset for managing images.

Details

class cutcutcodec.core.nn.dataset.image.ImageDataset(root: Path | str | bytes, shape: tuple[Integral, Integral] | list[Integral], *, dataaug: Callable[[FrameVideo], FrameVideo] | None = None, **kwargs)

A specific dataset for managing images.

Initialise and create the class.

Parameters

rootpathlike

Transmitted to Dataset initialisator.

shapeint and int

The pixel dimensions of the returned image. The image will be random reshaped and random cropped to reach this final shape. The convention adopted is the numpy convention (height, width).

dataaugcallable, optional

If provided, the function is called for each brut readed image before normalization.

**kwargsdict

Transmitted to Datset initialisator.