org.canova.image.recordreader

Class ImageNetRecordReader

java.lang.Object

org.canova.image.recordreader.BaseImageRecordReader

org.canova.image.recordreader.ImageNetRecordReader

All Implemented Interfaces:

Closeable, Serializable, AutoCloseable, org.canova.api.conf.Configurable, org.canova.api.records.reader.RecordReader

public class ImageNetRecordReader
extends BaseImageRecordReader

Record reader to handle ImageNet dataset Built to avoid changing api at this time. Api should change to track labels that are only referenced by id in filename Creates a hashmap for label name to id and references that with filename to generate matching lables.

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	eval
protected String	fileNameMapPath
protected Map<String,String>	labelFileIdMap
protected String	labelPath
protected static org.slf4j.Logger	log

Fields inherited from class org.canova.image.recordreader.BaseImageRecordReader

appendLabel, channels, CHANNELS, conf, CROP_IMAGE, cropImage, currentFile, fileNameMap, height, HEIGHT, hitImage, IMAGE_LOADER, imageLoader, inputSplit, iter, labels, pattern, patternPosition, record, width, WIDTH

Fields inherited from interface org.canova.api.records.reader.RecordReader

APPEND_LABEL, LABELS, NAME_SPACE

Constructor Summary

Constructors

Constructor and Description
ImageNetRecordReader(int height, int width, int channels, String labelPath)
ImageNetRecordReader(int height, int width, int channels, String labelPath, boolean appendLabel)
ImageNetRecordReader(int height, int width, int channels, String labelPath, boolean appendLabel, String pattern)
ImageNetRecordReader(int height, int width, int channels, String labelPath, boolean appendLabel, String pattern, int patternPosition)
ImageNetRecordReader(int height, int width, int channels, String labelPath, String fileNameMapPath, boolean appendLabel)
ImageNetRecordReader(int height, int width, int channels, String labelPath, String fileNameMapPath, boolean appendLabel, String pattern, int patternPosition)

Method Summary

Modifier and Type	Method and Description
void	initialize(org.canova.api.split.InputSplit split)
Collection<org.canova.api.writable.Writable>	next()
Collection<org.canova.api.writable.Writable>	record(URI uri, DataInputStream dataInputStream)

Methods inherited from class org.canova.image.recordreader.BaseImageRecordReader

accumulateLabel, close, containsFormat, getConf, getCurrentFile, getLabel, getLabels, hasNext, initialize, numLabels, reset, setConf, setCurrentFile

Methods inherited from class java.lang.Object

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

Field Detail

log

protected static org.slf4j.Logger log

labelFileIdMap

protected Map<String,String> labelFileIdMap

labelPath

protected String labelPath

fileNameMapPath

protected String fileNameMapPath

eval

protected boolean eval

Constructor Detail

ImageNetRecordReader

public ImageNetRecordReader(int height,
                            int width,
                            int channels,
                            String labelPath)

ImageNetRecordReader

public ImageNetRecordReader(int height,
                            int width,
                            int channels,
                            String labelPath,
                            boolean appendLabel)

ImageNetRecordReader

public ImageNetRecordReader(int height,
                            int width,
                            int channels,
                            String labelPath,
                            boolean appendLabel,
                            String pattern)

ImageNetRecordReader

public ImageNetRecordReader(int height,
                            int width,
                            int channels,
                            String labelPath,
                            boolean appendLabel,
                            String pattern,
                            int patternPosition)

ImageNetRecordReader

public ImageNetRecordReader(int height,
                            int width,
                            int channels,
                            String labelPath,
                            String fileNameMapPath,
                            boolean appendLabel)

ImageNetRecordReader

public ImageNetRecordReader(int height,
                            int width,
                            int channels,
                            String labelPath,
                            String fileNameMapPath,
                            boolean appendLabel,
                            String pattern,
                            int patternPosition)

Method Detail

initialize

public void initialize(org.canova.api.split.InputSplit split)
                throws IOException

Specified by:

initialize in interface org.canova.api.records.reader.RecordReader

Overrides:

initialize in class BaseImageRecordReader

Throws:

IOException

next

public Collection<org.canova.api.writable.Writable> next()

Specified by:

next in interface org.canova.api.records.reader.RecordReader

Overrides:

next in class BaseImageRecordReader

record

public Collection<org.canova.api.writable.Writable> record(URI uri,
                                                           DataInputStream dataInputStream)
                                                    throws IOException

Specified by:

record in interface org.canova.api.records.reader.RecordReader

Overrides:

record in class BaseImageRecordReader

Throws:

IOException