This model was trained on Cat--VS--Dogs dataset from kaggle
Link to the pretrained model
- $ git clone this repo
- download the dataset from given website
- extract the dataset
- replace path in the file with your path to petimages folder
- now open jupyter notebook
- open this file using jupyter notebook
- run the cells in the sequence they are in.
- $ pip install notebook <- this would install jupyter notebook
- $ jupyter notebook <- this would start notebook in local server
Layer | Shape | Params |
---|---|---|
Conv2D | (None, 98, 98, 32) | 896 |
MaxPooling2D | (None, 49, 49, 32) | 0 |
Conv2D | (None, 47, 47, 64) | 18496 |
MaxPooling2D | (None, 23, 23, 64) | 0 |
Flattenn | (None, 33856) | 0 |
Dense | (None, 64) | 2166848 |
Dense | (None, 1) | 65 |