def postprocess_data(self, datum):
    'Final postprocessing step before image is loaded into the batch.'
    return nd.transpose(datum, axes=(2, 0, 1))
Final postprocessing step before image is loaded into the batch.
def postprocess_data(self, datum):
    
    return nd.transpose(datum, axes=(2, 0, 1))