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)
- Downloads last month
- 4
Inference API (serverless) does not yet support generic models for this pipeline type.