Skip to content

kashishvjain/Image-Classifier-Web-App

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image_Classifier_Web_App

A website wherin a transfer learning model is deployed and classifies uploaded images and gives the option to change the hyperparameters of the model while retraining

The model used was vgg19

location:hackathon\imagesrec\static\imagesrec\vgg_model2

Transfer_Learning.ipynb

The jupyter notebook used for transfer learning on Google Colab to train the model

Dataset :

location:data(the link for this dataset is as follows: https://www.kaggle.com/puneet6060/intel-image-classification

Functionalities :

1.Bulk Training Model and Inference

2.Retraining model using the given images

3.Training the model by changing parameters

4.Integration django with pytorch

Explanation of our website

location:ppt_final.pdf

Screenshots

1.Home Page

2.Uploading Image Page

3.Results Page

4.Changing Hyperparameters Page

5.Plot of newly trained Model Page

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published