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

0 stars

Watchers

0 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors