org.nd4j.linalg.dataset.api.preprocessor

Class CropAndResizeDataSetPreProcessor

java.lang.Object

org.nd4j.linalg.dataset.api.preprocessor.CropAndResizeDataSetPreProcessor

All Implemented Interfaces:

Serializable, DataSetPreProcessor

public class CropAndResizeDataSetPreProcessor
extends Object
implements DataSetPreProcessor

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	CropAndResizeDataSetPreProcessor.ResizeMethod

Constructor Summary

Constructors

Constructor and Description
CropAndResizeDataSetPreProcessor(int originalHeight, int originalWidth, int cropYStart, int cropXStart, int resizedHeight, int resizedWidth, int numChannels, CropAndResizeDataSetPreProcessor.ResizeMethod resizeMethod)

Method Summary

Modifier and Type	Method and Description
void	preProcess(DataSet dataSet) NOTE: The data format must be NHWC

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

CropAndResizeDataSetPreProcessor

public CropAndResizeDataSetPreProcessor(int originalHeight,
                                        int originalWidth,
                                        int cropYStart,
                                        int cropXStart,
                                        int resizedHeight,
                                        int resizedWidth,
                                        int numChannels,
                                        CropAndResizeDataSetPreProcessor.ResizeMethod resizeMethod)

Parameters:

originalHeight - Height of the input datasets

originalWidth - Width of the input datasets

cropYStart - y coord of the starting point on the input datasets

cropXStart - x coord of the starting point on the input datasets

resizedHeight - Height of the output dataset

resizedWidth - Width of the output dataset

numChannels -

resizeMethod -

Method Detail

preProcess

public void preProcess(DataSet dataSet)

NOTE: The data format must be NHWC

Specified by:

preProcess in interface DataSetPreProcessor

Parameters:

dataSet - the data set to pre process