|
--- |
|
tags: |
|
- image-classification |
|
library_name: generic |
|
--- |
|
|
|
# Dog vs Cat Image Classification with FastAI CNN |
|
|
|
Training is based in FastAI [Quick Start](https://docs.fast.ai/quick_start.html). Example training |
|
|
|
|
|
## Training |
|
|
|
The model was trained as follows |
|
|
|
```python |
|
path = untar_data(URLs.PETS)/'images' |
|
|
|
def is_cat(x): return x[0].isupper() |
|
dls = ImageDataLoaders.from_name_func( |
|
path, get_image_files(path), valid_pct=0.2, seed=42, |
|
label_func=is_cat, item_tfms=Resize(224)) |
|
|
|
learn = cnn_learner(dls, resnet34, metrics=error_rate) |
|
learn.fine_tune(1) |
|
``` |
