ImageTimmDataset

class lightautoml.image.image.ImageTimmDataset(model, data, loader=<function pil_loader>)

Bases: object

Image for Timm Dataset Class.

__init__(model, data, loader=<function pil_loader>)

Pytorch Dataset for TimmModelEmbedder.

Parameters:

• model (TimmModelEmbedder) – model which we train.

• data (Sequence[str]) – Sequence of paths.

• loader (Callable) – Callable for reading image from path.