nikilasriram/pets-convnext-sagemaker

Pet breed classifier (37 cat and dog breeds)

Fine-tuned on the Oxford-IIIT Pet dataset (80/20 stratified split; photo counts in the table below) with a SageMaker training job, for a Distributed Computing class demo.

base model	facebook/convnext-tiny-224
epochs	5
validation accuracy	93.2%
trained on	SageMaker training job pets-pytorch-2026-11-10-18-16-14-787 (ml.g6.xlarge)
billable seconds	219
photos	1,480 training / 370 validation

Use it

from transformers import pipeline
classifier = pipeline("image-classification", model="nikilasriram/pets-convnext-sagemaker")
classifier("my_cat.jpg")

Breeds: Abyssinian, Bengal, Birman, Bombay, British Shorthair, Egyptian Mau, Maine Coon, Persian, Ragdoll, Russian Blue, Siamese, Sphynx, American Bulldog, American Pit Bull Terrier, Basset Hound, Beagle, Boxer, Chihuahua, English Cocker Spaniel, English Setter, German Shorthaired, Great Pyrenees, Havanese, Japanese Chin, Keeshond, Leonberger, Miniature Pinscher, Newfoundland, Pomeranian, Pug, Saint Bernard, Samoyed, Scottish Terrier, Shiba Inu, Staffordshire Bull Terrier, Wheaten Terrier, Yorkshire Terrier.

Dataset: Parkhi et al., "Cats and Dogs", CVPR 2012; images under CC BY-SA 4.0.