Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/nirmit27/dog-vs-cat-classification

A binary image classification model based on Transfer Learning for distinguishing images of dogs from cats.
https://github.com/nirmit27/dog-vs-cat-classification

binary-classification deep-learning glob image-classification image-processing jupyter-notebook machine-learning opencv opencv2 python tensorflow transfer-learning

Last synced: 6 days ago
JSON representation

A binary image classification model based on Transfer Learning for distinguishing images of dogs from cats.

Awesome Lists containing this project

README

        


Dog vs. Cat Image Classification


A binary image classification model based on Transfer Learning for distinguishing images of dogs from cats.


## Model
The pre-trained **MobileNet-v2** SSD-based object detection model trained on Open Images V4 with ImageNet pre-trained MobileNet V2 as image feature extractor, downloaded from the link down below.
> [MobileNet-v2 model](https://www.kaggle.com/models/google/mobilenet-v2/TensorFlow2/tf2-preview-feature-vector/4)

## Dataset
Accept the Challenge's "Terms and Conditions" in the **Rules** section before downloading the dataset from the link down below.
> [Kaggle Competitions dataset](https://www.kaggle.com/competitions/dogs-vs-cats-redux-kernels-edition/data)