image_output_1.png
base_dir = '../input/cell_images/cell_images/'
base_path = Path(base_dir)
base_path

data = ImageDataBunch.from_folder(base_path, valid_pct=0.1, train='.', ds_tfms=get_transforms(max_warp=0, flip_vert=True), size=128, bs=32, num_workers=0).normalize(imagenet_stats)
print(f'Classes to classify: \n {data.classes}')
data.show_batch(rows=5, figsize=(7, 7))