public abstract static class ObjectDetectionTranslator.ObjectDetectionBuilder<T extends ObjectDetectionTranslator.ObjectDetectionBuilder> extends BaseImageTranslator.ClassificationBuilder<T>
Modifier and Type | Field and Description |
---|---|
protected double |
imageHeight |
protected double |
imageWidth |
protected float |
threshold |
synsetLoader
batchifier, flag, height, pipeline, width
Constructor and Description |
---|
ObjectDetectionBuilder() |
Modifier and Type | Method and Description |
---|---|
protected void |
configPostProcess(java.util.Map<java.lang.String,?> arguments) |
double |
getImageHeight()
Get resized image height.
|
double |
getImageWidth()
Get resized image width.
|
T |
optRescaleSize(double imageWidth,
double imageHeight)
Sets the optional rescale size.
|
T |
optThreshold(float threshold)
Sets the threshold for prediction accuracy.
|
optSynset, optSynsetArtifactName, optSynsetUrl, validate
addTransform, configPreProcess, optBatchifier, optFlag, self, setPipeline
protected float threshold
protected double imageWidth
protected double imageHeight
public T optThreshold(float threshold)
Predictions below the threshold will be dropped.
threshold
- the threshold for the prediction accuracypublic T optRescaleSize(double imageWidth, double imageHeight)
imageWidth
- the width to rescale images toimageHeight
- the height to rescale images topublic double getImageWidth()
public double getImageHeight()
protected void configPostProcess(java.util.Map<java.lang.String,?> arguments)
configPostProcess
in class BaseImageTranslator.ClassificationBuilder<T extends ObjectDetectionTranslator.ObjectDetectionBuilder>