image_output_1.png
path = Path('../input/heroes/heroes/')
path.ls()

data = ImageDataBunch.from_folder(path, train='.', valid_pct=0.1, ds_tfms=get_transforms(max_warp=0, flip_vert=True, do_flip=True), size=128, bs=16).normalize(imagenet_stats)
print(f'Classes: \n {data.classes}')
data.show_batch(rows=8, figsize=(10, 10))