Skip to content

Python script for a command line app to build and train a neural network using either VGG19 or AlexNet architecture. The app classifies images into 102 categories of flowers, and it provides flexibility to use custom data for training, enabling users to categorize other objects as well. The implementation leverages transfer learning.

License

Notifications You must be signed in to change notification settings

ak586/ImageClassifier

Repository files navigation

AI Programming with Python Project

Project code for Udacity's AI Programming with Python Nanodegree program. In this project, students first develop code for an image classifier built with PyTorch, then convert it into a command line application.

About

Python script for a command line app to build and train a neural network using either VGG19 or AlexNet architecture. The app classifies images into 102 categories of flowers, and it provides flexibility to use custom data for training, enabling users to categorize other objects as well. The implementation leverages transfer learning.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published