Search…
IMAGE_TO_NDARRAY
ImageToNDArrayStep is a PipelineStep for converting images to n-dimensional arrays. The exact way that images are converted is highly configurable (formats, channels, output sizes, normalization, etc).
Configs
Descriptions
config
Configuration 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 n-dimensional array.
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. Not relevant if metadata == false. Default is @ImageToNDArrayStepMetadata
ImageToNDArrayConfig is configuration for converting an image into n-dimensional array. This configuration is used in config from ImageToNDArrayStep, for example:
inferenceConfiguration.pipeline(SequencePipeline.builder()
.add(new ImageToNDArrayStep() //add ImageToNDArrayStep() into pipeline to set image to NDArray for input
.config(new ImageToNDArrayConfig() //image configuration
.width(28)
.height(28)
.dataType(NDArrayType.FLOAT)
.aspectRatioHandling(AspectRatioHandling.CENTER_CROP)
.includeMinibatchDim(true)
.channelLayout(NDChannelLayout.GRAYSCALE)
.format(NDFormat.CHANNELS_FIRST)
.normalization(ImageNormalization.builder().type(ImageNormalization.Type.SCALE).build())
)
.keys("image")
.outputNames("input_layer")
.keepOtherValues(true)
.metadata(false)
.metadataKey(ImageToNDArrayStep.DEFAULT_METADATA_KEY))
.build()
Configs
Descriptions
height
Output array image height. Leave null to convert to the same size as the image height.
width
Output array image width. Leave null to convert to the same size as the image width.
dataType
Data type of the n-dimensional array. Default value is FLOAT.
includeMinibatchDim
If true, the output array will contain an extra dimension for the minibatch number. This will look like (1, Channels, Height, Width) instead of (Channels, Height, Width) for format == CHANNELS_FIRST or (1, Height, Width, Channels) instead of (Height, Width, Channels) for format == CHANNELS_LAST. Default is true.
aspectRatioHandling
An enum to Handle the situation where the input image and output NDArray have different aspect ratios.
CENTER_CROP (crop larger dimension then resize if necessary), PAD (pad smaller dimension then resize if necessary), STRETCH (simply resize, distorting if necessary). Default is CENTER_CROP.
format
The format to be used when converting an Image to an NDArray. Default is CHANNEL_FIRST. Another option is CHANNEL_LAST.
channelLayout
An enum that represents the type (and order) of the color channels for an image after it has been converted to an NDArray. For example, RGB vs. BGR etc, default value is RGB.
normalization
Configuration that specifies the normalization type of an image array values.
listHandling
An enum to specify how to handle a list of input images. Default is NONE.
Copy link