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.
-
Notifications
You must be signed in to change notification settings - Fork 0
Gazalshyam/Image_Classifier
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Image Classifier Project for AWS AI and ML Scholarship Program
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
