(self, dataset_name, img_res=(128, 128))
| 6 | matplotlib.use('Agg') |
| 7 | class DataLoader(): |
| 8 | def __init__(self, dataset_name, img_res=(128, 128)): |
| 9 | self.dataset_name = dataset_name |
| 10 | self.img_res = img_res |
| 11 | |
| 12 | def load_data(self, batch_size=1, is_testing=False): |
| 13 | data_type = "train" if not is_testing else "test" |
nothing calls this directly
no outgoing calls
no test coverage detected