weka.dl4j

Class EasyImageRecordReader

java.lang.Object

org.datavec.api.records.reader.BaseRecordReader

org.datavec.image.recordreader.BaseImageRecordReader

weka.dl4j.EasyImageRecordReader

All Implemented Interfaces:

java.io.Closeable, java.io.Serializable, java.lang.AutoCloseable, org.datavec.api.conf.Configurable, org.datavec.api.records.reader.RecordReader

public class EasyImageRecordReader
extends org.datavec.image.recordreader.BaseImageRecordReader

ImageRecordReader in DeepLearning4j assumes that your images are separated into different folders, where each folder is a class, e.g. for MNIST, all the 0 images are in a folder called 0/, all the 1 images in 1/, etc. At test time, you will get instances where the class is not known, e.g:
image.png,?
This is why we have had to make a new image record reader and make it possible that all the images are in one folder. We specify the classes for those images explicitly. The reader also shuffles the data.

Version:

$Revision: 11711 $

Author:

Chris Beckham

See Also:

Serialized Form

Field Summary

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

CHANNELS, CROP_IMAGE, HEIGHT, IMAGE_LOADER, WIDTH

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

APPEND_LABEL, LABELS, NAME_SPACE

Constructor Summary

Constructors

Constructor and Description
EasyImageRecordReader(int width, int height, int channels, java.util.ArrayList<java.io.File> filenames, java.util.ArrayList<java.lang.String> classes, int seed) Constructor for the record reader.

Method Summary

Modifier and Type	Method and Description
void	initialize(org.datavec.api.split.InputSplit split) Initializes the iterators.
java.util.List<org.datavec.api.writable.Writable>	next() Gets the next element
java.util.List<org.datavec.api.writable.Writable>	record(java.net.URI uri, java.io.DataInputStream dataInputStream) A method that still needs to be implemented.
void	reset() Resets the record reader.

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

close, getConf, getCurrentFile, getLabel, getLabels, hasNext, initialize, initialize, initialize, loadFromMetaData, loadFromMetaData, nextRecord, numLabels, setConf, setCurrentFile, setLabels

Methods inherited from class org.datavec.api.records.reader.BaseRecordReader

getListeners, setListeners, setListeners

Methods inherited from class java.lang.Object

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

Constructor Detail

EasyImageRecordReader

public EasyImageRecordReader(int width,
                             int height,
                             int channels,
                             java.util.ArrayList<java.io.File> filenames,
                             java.util.ArrayList<java.lang.String> classes,
                             int seed)
                      throws java.io.IOException

Constructor for the record reader.

Parameters:

width - The desired image width.

height - The desired image height.

channels - The desired number of channels.

filenames - The list of file names.

classes - The list of class values.

seed - the seed for the random number generator.

Throws:

java.io.IOException

Method Detail

initialize

public void initialize(org.datavec.api.split.InputSplit split)

Initializes the iterators. Shuffles the lists first.

Specified by:

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

Overrides:

initialize in class org.datavec.image.recordreader.BaseImageRecordReader

Parameters:

split - this parameter is ignored.

next

public java.util.List<org.datavec.api.writable.Writable> next()

Gets the next element

Specified by:

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

Overrides:

next in class org.datavec.image.recordreader.BaseImageRecordReader

Returns:

the next element

reset

public void reset()

Resets the record reader.

Specified by:

reset in interface org.datavec.api.records.reader.RecordReader

Overrides:

reset in class org.datavec.image.recordreader.BaseImageRecordReader

record

public java.util.List<org.datavec.api.writable.Writable> record(java.net.URI uri,
                                                                java.io.DataInputStream dataInputStream)
                                                         throws java.io.IOException

A method that still needs to be implemented.

Specified by:

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

Overrides:

record in class org.datavec.image.recordreader.BaseImageRecordReader

Parameters:

uri - the URI

dataInputStream - the data input stream

Returns:

the record

Throws:

java.io.IOException