def load_data(dataset, p=PATH.paths, input_shape=(128, 128, 3)):
if dataset == "dataset":
data = ImgData.CancerData(p["dataset"], input_shape)
if dataset == "dataset2018":
data = ImgData.CancerData2018(p["dataset2018"], input_shape)
if dataset == "cifar10":
data = ImgData.Cifar10Data(p["cifar10"], input_shape)
data.load()
return data
