ImageDataset

class lightautoml.image.image.ImageDataset(data, is_advprop=True, loader=<function pil_loader>)[source]

Bases: object

Image Dataset Class.

__init__(data, is_advprop=True, loader=<function pil_loader>)[source]

Pytorch Dataset for EffNetImageEmbedder.

Parameters

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

  • is_advprop (bool) – Use adversarial training.

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