Downloads · 30 days
5
3% of all-time downloads
cat_vs_dog_v2
AustinL/cat_vs_dog_v2
cat_vs_dog_v2 is a image classification model from AustinL. 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
192
Public
Repo size
87.5 MB
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
- Aug 19, 2023
- Updated
- Aug 19, 2023
- SHA
- 77581ef4
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
- Aug 19, 2023
- Updated
- Aug 19, 2023