Skip to content

KaushikJais/Flower-Classifier-Using-pytorch-and-TransferLearning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

Flower-Classifier-Using-pytorch-and-TransferLearning

Trained a resnet50 model to classify flower species from the Oxford Flower Dataset
20 epochs with frozen pretrained layers, 7 epochs with all layers unfrozen
Optimizer - Adam with Learning Rate 0.001 when layers were frozen and 0.0001 when layers are unfrozen
Accuracy on Test Dataset - 97.8%

Checkpoint not uploaded due to file-size limits set by github.

About

Build an Image Classifier(Flower species Classifier) using Tranfer Learning in Pytorch as part of Pytorch scholarship term

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published