R Deep Learning Cookbook

How to do it...

1. Load the .rec file as iterators. The following is the function to load the .rec data as iterators:

# Function to load data as iterators 
data.iterator <- function(data.shape, train.data, val.data, BATCHSIZE = 128) { 
   
  # Load training data as iterator 
  train <- mx.io.ImageRecordIter( 
    path.imgrec = train.data, 
    batch.size  = BATCHSIZE, 
    data.shape  = data.shape, 
    rand.crop   = TRUE, 
    rand.mirror = TRUE) 
   
  # Load validation data as iterator 
  val <- mx.io.ImageRecordIter( 
    path.imgrec = val.data, 
    batch.size  = BATCHSIZE, 
    data.shape  = data.shape, 
    rand.crop   = FALSE, 
    rand.mirror = FALSE 
  ) 
   
  return(list(train = train, val = val)) 
} 

In the preceding function, mx.io.ImageRecordIter reads batches of images from the RecordIO (.rec) files.

2. Load data ...