Downloads · 30 days
9
1% of all-time downloads
fastai_cat_vs_dog
osanseviero/fastai_cat_vs_dog
fastai_cat_vs_dog is a image classification model from osanseviero. Use it when you need a label for an image. It is set up for generic.
Training is based in FastAI Quick Start. Example training
All-time downloads
1.1K
Public
Repo size
175 MB
Likes
4
Public
Repo makeup
Click a slice to open those files.
.pkl87.5 MB · 100%
At a glance
- Task
- Image Classification
- Library
- generic
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Dec 17, 2021
- SHA
- cc3aeea1
Model card
From the Hugging Face model README
Dog vs Cat Image Classification with FastAI CNN
Training is based in FastAI Quick Start. Example training
Training
The model was trained as follows
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)
- Task
- Image Classification
- Library
- generic
- Created
- Mar 2, 2022
- Updated
- Dec 17, 2021