Skip to content

Image Classifier Project for AWS AI and ML Scholarship Program

Notifications You must be signed in to change notification settings

Gazalshyam/Image_Classifier

Repository files navigation

In this project, I implemented an image classifier with PyTorch. A pre-trained VGG16 model was loaded, and its parameters were frozen. The parameters of the feedforward classifier were appropriately trained, while the parameters of the feature network were left static. This model allows users to choose from three different models. It is compatible for users without GPU.

image

About

Image Classifier Project for AWS AI and ML Scholarship Program

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published