public class CropAndResizeDataSetPreProcessor extends Object implements DataSetPreProcessor
Modifier and Type | Class and Description |
---|---|
static class |
CropAndResizeDataSetPreProcessor.ResizeMethod |
Constructor and Description |
---|
CropAndResizeDataSetPreProcessor(int originalHeight,
int originalWidth,
int cropYStart,
int cropXStart,
int resizedHeight,
int resizedWidth,
int numChannels,
CropAndResizeDataSetPreProcessor.ResizeMethod resizeMethod) |
Modifier and Type | Method and Description |
---|---|
void |
preProcess(DataSet dataSet)
NOTE: The data format must be NHWC
|
public CropAndResizeDataSetPreProcessor(int originalHeight, int originalWidth, int cropYStart, int cropXStart, int resizedHeight, int resizedWidth, int numChannels, CropAndResizeDataSetPreProcessor.ResizeMethod resizeMethod)
originalHeight
- Height of the input datasetsoriginalWidth
- Width of the input datasetscropYStart
- y coord of the starting point on the input datasetscropXStart
- x coord of the starting point on the input datasetsresizedHeight
- Height of the output datasetresizedWidth
- Width of the output datasetnumChannels
- resizeMethod
- public void preProcess(DataSet dataSet)
preProcess
in interface DataSetPreProcessor
dataSet
- the data set to pre processCopyright © 2020. All rights reserved.