Skip to content

Latest commit

 

History

History
7 lines (4 loc) · 474 Bytes

README.md

File metadata and controls

7 lines (4 loc) · 474 Bytes

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.