| from fastai.vision.all import * | |
| path = untar_data(URLs.PETS) | |
| dls = ImageDataLoaders.from_name_re(path, | |
| get_image_files(path/'images'), pat='(.+)_\d+.jpg', | |
| item_tfms=Resize(460), | |
| batch_tfms=aug_transforms(size=224, min_scale=0.75), | |
| num_workers=0, | |
| bs=16, | |
| ) | |
| learn = vision_learner(dls, models.resnet50, metrics=accuracy) | |
| learn.fine_tune(1) | |
| learn.path = Path('.') | |
| learn.export() |