ImageFeaturesTransformer

class lightautoml.transformers.image.ImageFeaturesTransformer(hist_size=30, is_hsv=True, n_jobs=4, loader=<function pil_loader>)[source]

Bases: LAMLTransformer

Simple image histogram.

__init__(hist_size=30, is_hsv=True, n_jobs=4, loader=<function pil_loader>)[source]

Create normalized color histogram for rgb or hsv image.

Parameters:
  • hist_size (int) – Number of bins for each channel.

  • is_hsv (bool) – Convert image to hsv.

  • n_jobs (int) – Number of threads for multiprocessing.

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

property features

Features list.

Returns:

List of features names.

fit(dataset)[source]

Init hist class and create feature names.

Parameters:

dataset (Union[NumpyDataset, PandasDataset]) – Pandas or Numpy dataset of text features.

Returns:

self.

transform(dataset)[source]

Transform image dataset to color histograms.

Parameters:

dataset (Union[NumpyDataset, PandasDataset]) – Pandas or Numpy dataset of image paths.

Return type:

NumpyDataset

Returns:

Dataset with encoded text.