Skip to content

AyushDhanai1419/BirdClassifier_Pytorch_Side_Project

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

40 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

BirdClassifier_Pytorch_Side_Project

Contibuters

  • Ayush Dhanai - Slack Id : @Ayush Dhanai
  • Akhil Tiwari    - Slack Id : @Akhil_Tiwari

Architecture and details of the model

  • Pre-Trained Model used : Densenet121
  • Dataset Used : Caltech-UCSD Birds 200
  • Accuracy Achieved : 94.5%
  • Optimizer Used : ADAM
  • Loss Function : NLLLoss
  • Learning Rate Scheduler : StepLR

Libraries and Frameworks Used

Dependencies

  1. Flask
  2. torch and torchvision
  3. numpy
  4. Pillow
  5. wikipedia

Running this project

  1. Clone or download the repository on your local machine
  2. Make sure all the dependencies are installed
  3. open the teminal and change current working directory to project directory
  4. set the value of FLASK_APP environment variable to "application.py"
  5. run command "flask run" inside the terminal

About the project -

Creating a web application with the help of convolutional neural networks and transfer learning, that can be used to detect the species of the bird from its image. Application can be proved useful in the field of Ornithology (branch of zoology concern with the study of birds) and can also be used by people interested in wildlife photography.

  Screenshots of working application


  Homepage -upload the image from your device into the application and then model will predict   the species of the bird and display information regarding that particular species.



  Displaying result

  Let us know if any issue arise and we'd love your contribution in making this project even better.

About

Creating a bird classifier using transfer learning on densenet121 using PyTorch.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published