Class ImageToNDArrayStep
- java.lang.Object
-
- ai.konduit.serving.data.image.step.ndarray.ImageToNDArrayStep
-
- All Implemented Interfaces:
ai.konduit.serving.pipeline.api.step.PipelineStep,ai.konduit.serving.pipeline.api.TextConfig,java.io.Serializable
public class ImageToNDArrayStep extends java.lang.Object implements ai.konduit.serving.pipeline.api.step.PipelineStepA PipelineStep for convertingImages toNDArrays. The exact way that images are converted is highly configurable (formats, channels, output sizes, normalization, etc) - seeImageToNDArrayConfigfor more details
The following configuration is available:- config: the
ImageToNDArrayConfigconfiguration for how conversion should be performed - keys: may be null. If non-null: These are the names of images in the Data instance to convert
- outputNames: may be null. If non-null: the input images are renamed to this in the output Data instance after conversion to NDArray
- keepOtherValues: True by default. If true: copy all the other (non-converted/non-image) entries in the input Data to the output Data
- metadata: False by default. If true: include metadata about the images in the output Data - for example if/how it was cropped, and the original input size.
- metadataKey: Sets the key that the metadata will be stored under. Default:
DEFAULT_METADATA_KEY. Not relevant if metadata == false
Note that metadata will have the following format:
If a single image is converted, the metadata Data instance will have a nested Data instance i.e.:Data meta = myData.get( "@ImageToNDArrayStepMetadata");}- Author:
- Alex Black
- See Also:
- Serialized Form
-
-
Field Summary
Fields Modifier and Type Field Description static java.lang.StringDEFAULT_METADATA_KEYstatic java.lang.StringMETA_CROP_REGIONstatic java.lang.StringMETA_IMG_Hstatic java.lang.StringMETA_IMG_Wstatic java.lang.StringMETA_INNAME_KEYstatic java.lang.StringMETA_OUTNAME_KEY
-
Constructor Summary
Constructors Constructor Description ImageToNDArrayStep(ImageToNDArrayConfig config, java.util.List<java.lang.String> keys, java.util.List<java.lang.String> outputNames, boolean keepOtherValues, boolean metadata, java.lang.String metadataKey)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description ImageToNDArrayStepkeys(java.lang.String... keys)ImageToNDArrayStepoutputNames(java.lang.String... outputNames)
-
-
-
Field Detail
-
DEFAULT_METADATA_KEY
public static final java.lang.String DEFAULT_METADATA_KEY
- See Also:
- Constant Field Values
-
META_INNAME_KEY
public static final java.lang.String META_INNAME_KEY
- See Also:
- Constant Field Values
-
META_OUTNAME_KEY
public static final java.lang.String META_OUTNAME_KEY
- See Also:
- Constant Field Values
-
META_IMG_H
public static final java.lang.String META_IMG_H
- See Also:
- Constant Field Values
-
META_IMG_W
public static final java.lang.String META_IMG_W
- See Also:
- Constant Field Values
-
META_CROP_REGION
public static final java.lang.String META_CROP_REGION
- See Also:
- Constant Field Values
-
-
Constructor Detail
-
ImageToNDArrayStep
public ImageToNDArrayStep(ImageToNDArrayConfig config, java.util.List<java.lang.String> keys, java.util.List<java.lang.String> outputNames, boolean keepOtherValues, boolean metadata, java.lang.String metadataKey)
-
-
Method Detail
-
outputNames
public ImageToNDArrayStep outputNames(java.lang.String... outputNames)
-
keys
public ImageToNDArrayStep keys(java.lang.String... keys)
-
-