public class ObjectDetectionRecordReader extends BaseImageRecordReader
Format of returned values: 4d array, with dimensions [minibatch, 4+C, h, w] Where the image is quantized into h x w grid locations.
Note that this matches the format required for Deeplearning4j's Yolo2OutputLayer
| Modifier and Type | Field and Description |
|---|---|
protected Image |
currentImage |
appendLabel, channels, CHANNELS, conf, CROP_IMAGE, cropImage, currentFile, fileNameMap, finishedInputStreamSplit, height, HEIGHT, hitImage, IMAGE_LOADER, imageLoader, imageTransform, inputSplit, iter, labelGenerator, labelMultiGenerator, labels, logLabelCountOnInit, pattern, patternPosition, record, width, WIDTH, writeLabellistenersAPPEND_LABEL, LABELS, NAME_SPACE| Constructor and Description |
|---|
ObjectDetectionRecordReader(int height,
int width,
int channels,
int gridH,
int gridW,
ImageObjectLabelProvider labelProvider) |
ObjectDetectionRecordReader(int height,
int width,
int channels,
int gridH,
int gridW,
ImageObjectLabelProvider labelProvider,
ImageTransform imageTransform)
When imageTransform != null, object is removed if new center is outside of transformed image bounds.
|
| Modifier and Type | Method and Description |
|---|---|
void |
initialize(InputSplit split)
Called once at initialization.
|
java.util.List<Writable> |
next()
Get the next record
|
java.util.List<java.util.List<Writable>> |
next(int num)
This method will be used, if batchesSupported() returns true.
|
Record |
nextRecord()
Similar to
RecordReader.next(), but returns a Record object, that may include metadata such as the source
of the data |
java.util.List<Writable> |
record(java.net.URI uri,
java.io.DataInputStream dataInputStream)
Load the record from the given DataInputStream
Unlike
RecordReader.next() the internal state of the RecordReader is not modified
Implementations of this method should not close the DataInputStream |
accumulateLabel, batchesSupported, close, containsFormat, getConf, getCurrentFile, getLabel, getLabels, hasNext, initialize, initialize, initialize, loadFromMetaData, loadFromMetaData, numLabels, reset, resetSupported, setConf, setCurrentFile, setLabelsgetListeners, invokeListeners, setListeners, setListenersprotected Image currentImage
public ObjectDetectionRecordReader(int height,
int width,
int channels,
int gridH,
int gridW,
ImageObjectLabelProvider labelProvider)
height - Height of the output imageswidth - Width of the output imageschannels - Number of channels for the output imagesgridH - Grid/quantization size (along height dimension) - Y axisgridW - Grid/quantization size (along height dimension) - X axislabelProvider - ImageObjectLabelProvider - used to look up which objects are in each imagepublic ObjectDetectionRecordReader(int height,
int width,
int channels,
int gridH,
int gridW,
ImageObjectLabelProvider labelProvider,
ImageTransform imageTransform)
height - Height of the output imageswidth - Width of the output imageschannels - Number of channels for the output imagesgridH - Grid/quantization size (along height dimension) - Y axisgridW - Grid/quantization size (along height dimension) - X axislabelProvider - ImageObjectLabelProvider - used to look up which objects are in each imageimageTransform - ImageTransform - used to transform image and coordinatespublic java.util.List<Writable> next()
RecordReadernext in interface RecordReadernext in class BaseImageRecordReaderpublic void initialize(InputSplit split) throws java.io.IOException
RecordReaderinitialize in interface RecordReaderinitialize in class BaseImageRecordReadersplit - the split that defines the range of records to readjava.io.IOExceptionpublic java.util.List<java.util.List<Writable>> next(int num)
RecordReadernext in interface RecordReadernext in class BaseImageRecordReaderpublic java.util.List<Writable> record(java.net.URI uri, java.io.DataInputStream dataInputStream) throws java.io.IOException
RecordReaderRecordReader.next() the internal state of the RecordReader is not modified
Implementations of this method should not close the DataInputStreamrecord in interface RecordReaderrecord in class BaseImageRecordReaderjava.io.IOException - if error occurs during reading from the input streampublic Record nextRecord()
RecordReaderRecordReader.next(), but returns a Record object, that may include metadata such as the source
of the datanextRecord in interface RecordReadernextRecord in class BaseImageRecordReader