Image Loading

test.java

package com.chil;


import org.canova.api.records.reader.RecordReader;
import org.canova.api.split.FileSplit;
import org.canova.image.recordreader.ImageRecordReader;
import org.deeplearning4j.datasets.canova.RecordReaderDataSetIterator;

import org.nd4j.linalg.dataset.DataSet;
import org.nd4j.linalg.dataset.api.iterator.DataSetIterator;
import java.io.File;
import java.util.ArrayList;
import java.util.List;


/**
 * Hello world!
 *
 */
public class App 
{
    public static void main( String[] args ) throws Exception
    {
        String labeledPath = System.getProperty("user.home")+"/test";
        List<String> labels = new ArrayList<String>();

        //traverse dataset to get each label
        for(File f : new File(labeledPath).listFiles())
        {
         labels.add(f.getName());
        }
        RecordReader recordReader = new ImageRecordReader(28, 28,3, true, labels);
        recordReader.initialize(new FileSplit(new File(labeledPath)));
        DataSetIterator iter = new RecordReaderDataSetIterator(recordReader, 784*3,labels.size());
        Integer i=1;
        while(iter.hasNext()){
            DataSet next = iter.next();
            System.out.println("Reading  "+i+" = "+next.getFeatureMatrix()+" added");
        }
        System.out.println( "Hello World!  iter is "+recordReader );
    }
}