ImageFeaturesTransformer

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

Bases: lightautoml.transformers.base.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.

Return type: List[str]
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.