Skip to content

krunalvaghani/image_classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image Classification

Download the dataset from https://www.kaggle.com/tongpython/cat-and-dog and place it in the folder training_set and test_set

Use Image_classification.py to train the classifier, developed using VGG16 architecture. You can use this script to do multi class classifiction as well. For the multiclass classification, do the changes in scripts where it alreday mentioned.

It also provides script for manual data generator. Usage described in the file.