Skip to content

Conv Neural Network Image Classifier using torchvision pretrained models to train a classifier while preserving the features.

Notifications You must be signed in to change notification settings

conwubiko/AI-Image-Classifier

Repository files navigation

AI-Image-Classifier

Uses Deep Conv Neural Network for Large-Scale Image Classifier in python, pytorch and torchvision. This project uses deep learning model pretrained on hundreds of thousands of images as part of the overall application architecture. It uses and compares VGG13, VGG19, Densenet, Resnet152 FC for transfer learning in order to predict different types of flowers. The application is portable over to a mobile device e.g. phone in order to capture images of unknown flowers to make accurate predictions of their names and types.

Usage

There are four files to use, one is the train.py, and the other is the predict.py, and two utility files, namely arguments and utils.

About

Conv Neural Network Image Classifier using torchvision pretrained models to train a classifier while preserving the features.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published