Java Deep Learning Cookbook

How to do it...

1. Load and initialize the training data using ImageRecordReader:

ImageRecordReader imageRecordReader = new ImageRecordReader(imageHeight,imageWidth,channels,parentPathLabelGenerator); imageRecordReader.initialize(trainData,null);

2. Create a dataset iterator using RecordReaderDataSetIterator:

DataSetIterator dataSetIterator = new RecordReaderDataSetIterator(imageRecordReader,batchSize,1,numLabels);

3. Add the normalizer to the dataset iterator:

DataNormalization scaler = new ImagePreProcessingScaler(0,1); scaler.fit(dataSetIterator); dataSetIterator.setPreProcessor(scaler); 

4. Train the model by calling fit():

MultiLayerConfiguration config = builder.build(); MultiLayerNetwork model = new MultiLayerNetwork(config); model.init(); ...