Downloads · 30 days
13
2% of all-time downloads
fastai_cat_vs_dog_fork2
osanseviero/fastai_cat_vs_dog_fork2
fastai_cat_vs_dog_fork2 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
628
Public
Repo size
—
Likes
0
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