def load_data(dataset, p=PATH.paths, input_shape=(128, 128, 3)): if dataset == "dataset": data = ImgData.CancerData(p["dataset"], input_shape) else: data = ImgData.CancerData(p["cifar10"], input_shape) data.load() # check_data(data) return data
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