slsnebur / cnn-image-classifier

Solving (binary) image classification problem using convolutional neural network

cnn-image-classifier

https://www.youtube.com/watch?v=py5byOOHZM8&t=13m56s

Pretrained models and augmented dataset: [MEGA]

Pretrained models only: [MEGA]

.
├── README.md
├── datasets
│   └── train
│       ├── cat.0.jpg
│       ├── cat.2.jpg
│       ├── dog.1640.jpg
│       ├── dog.1641.jpg
│       └── dog.1642.jpg
├── predict
│   ├── A.jpg
│   ├── B.jpg
│   ├── C.jpg
│   ├── D.jpg
│   ├── E.jpg
│   └── F.jpg
└── src
    └── cic
        ├── dog-cat-sc50-train.npy
        ├── model1
        │   ├── accuracy.png
        │   ├── loss.png
        │   └── model1.model
        ├── model2
        │   ├── accuracy.png
        │   ├── loss.png
        │   └── model2.model
        ├── model3
        │   ├── accuracy.png
        │   ├── loss.png
        │   └── model3.model
        ├── predict.py
        └── train.py