loader module

class loader.Loader(parser, config)

Read in HDF5 files from train and test directory and perform some checks to make sure everything is okay.

Parameters

parser (argparse object): Command line arguments handled by argparse.

config (dict): Dictionary of configuration which was made from YAML.

Methods

getImageCountsPerFile()	Returns
getMax()	Returns
getMin()	Returns
getTestingFiles()	Returns
getTestingH5Files()	Returns
getTotalImages()	Returns
getTrainingFiles()	Returns
getTrainingH5Files()	Returns
getXShape()	Returns
getYShape()	Returns
prepareData(files[, test])	Check the shapes across all data sets to make sure all is good.
readDir(dir_name)	Read all files in a directory.

getImageCountsPerFile()

Returns

A dict with training filenames as keys and number of images as values.

getMax()

Returns

The maximim value of training and testing.

getMin()

Returns

The minimum value of training and testing.

getTestingFiles()

Returns

The list of testing filenames.

getTestingH5Files()

Returns

The list of testing h5py file objects.

getTotalImages()

Returns

The total number of images used in training.

getTrainingFiles()

Returns

The list of training filenames.

getTrainingH5Files()

Returns

The list of training h5py file objects.

getXShape()

Returns

The shape of the input data.

getYShape()

Returns

The shape of the output data.

prepareData(files, test=False)

Check the shapes across all data sets to make sure all is good.

Parameters

files (list): list of files to check.

test (bool): Set to true if they are testing files. Default: False.

Returns

A list of h5py file objects.

readDir(dir_name)

Read all files in a directory. This is called in the __init__ function.

Parameters

dir_name (str): Path in which files will be collected.

Returns

A list of files in the supplied path.