Skip to content

General CNN Classifier. A simple transfer learning example to classify images. The base model used is Xception. The code is generalized to work with any type of images and any number of classes.

Notifications You must be signed in to change notification settings

Its-a-me-Ashwin/General-CNN-Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

General-CNN-Classifier

General CNN Classifier

Path given to the function should cointain n folders (n classes), these folders should cointain pictures belonging to that perticular class. The program will train on these images and save the model as a .model file. This file can be rloaded later to be used for prediction.

For Eg: D:/DataSets/PetImages

PetImages has the follwing folders. Dogs bigDog.png helloDog.jpg .... Cats bigCat.png hellhelloCat.jpg .... Parrots parrot.png newparrot.jpg .... ...

About

General CNN Classifier. A simple transfer learning example to classify images. The base model used is Xception. The code is generalized to work with any type of images and any number of classes.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages