%reload_ext autoreload
%autoreload 2
%matplotlib inline

from fastai import *
from fastai.vision import *

path = untar_data(URLs.PETS)

tfms = get_transforms()
bs = 64

data = (InputList.from_folder(path/'images')
        .label_from_re(r'^(.*)_\d+.jpg$')
        .random_split_by_pct(0.2)
        .datasets(ImageClassificationDataset)
        .transform(tfms, size=224)
        .databunch(bs=bs, path=path))

data.show_batch(rows=3, figsize=(7,7))

data.path

' '.join(data.classes)

path = untar_data(URLs.PLANET_SAMPLE)

data = (InputList.from_folder(path)
        .label_from_csv('labels.csv', sep=' ', suffix='.jpg', folder='train')
        .random_split_by_pct(0.2)
        .datasets(ImageMultiDataset)
        .transform(tfms, size=128)
        .databunch(bs=bs, path=path))

data.show_batch(rows=3)

' '.join(data.classes)

path = Path('data/dogscats')

data = (InputList.from_folder(path)
        .label_from_folder()
        .split_by_folder()
        .datasets(ImageClassificationDataset)
        .transform(tfms, size=224)
        .databunch(bs=bs, path=path))

data.show_batch(rows=3)

data.classes

data.train_ds.x

data.valid_ds.x