public class TinyImageNetDataSetIterator extends RecordReaderDataSetIterator
RecordReaderDataSetIterator.Builder| Modifier and Type | Field and Description |
|---|---|
protected DataSetPreProcessor |
preProcessor |
batchNum, batchSize, converter, labelIndex, labelIndexTo, last, maxNumBatches, numPossibleLabels, recordReader, regression, sequenceIter, useCurrent| Constructor and Description |
|---|
TinyImageNetDataSetIterator(int batchSize) |
TinyImageNetDataSetIterator(int batchSize,
DataSetType set) |
TinyImageNetDataSetIterator(int batchSize,
int[] imgDim,
DataSetType set) |
TinyImageNetDataSetIterator(int batchSize,
int[] imgDim,
DataSetType set,
ImageTransform imageTransform,
long rngSeed)
Get the Tiny ImageNet iterator with specified train/test set and custom transform.
|
asyncSupported, batch, getLabels, hasNext, inputColumns, loadFromMetaData, loadFromMetaData, next, next, remove, reset, resetSupported, setCollectMetaData, setPreProcessor, totalOutcomesclone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, waitgetPreProcessorprotected DataSetPreProcessor preProcessor
public TinyImageNetDataSetIterator(int batchSize)
public TinyImageNetDataSetIterator(int batchSize,
DataSetType set)
public TinyImageNetDataSetIterator(int batchSize,
int[] imgDim,
DataSetType set)
public TinyImageNetDataSetIterator(int batchSize,
int[] imgDim,
DataSetType set,
ImageTransform imageTransform,
long rngSeed)
batchSize - Size of each patchimgDim - Dimensions of desired output - for example, {64, 64}set - Train, test, or validationimageTransform - Additional image transform for outputrngSeed - random number generator seed to use when shuffling examples