Skip to content

Volviane/Cats_Dogs_classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

27 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Cats_Dogs_classification

Dogs and Cats Classification with Convolutional Neural Network (CNN) Dog vs Cat Classification (2-CNN,4-CNN,Tranfer learning with pytorch and Tensorflow Keras)

Saphir Volviane Mfogo

Table of Contents

Introduction

This respository contains the Dog vs Cat classifier with regularization both with TensorFlow and Pytorch

Usage

Assuming you are in the Cats_Dogs_classification master folder the you go in Dog_Cat_Classifier subfolder. You also do the same for tranfer learning subfolder

  1. Preparation:
  • Download the Dog vs Cat dataset
  • Then put them into a folder (Cat_Dog_data/ for example). Please carefully check the directory of the folder whether containing invalid characters or image with wrong extension. The data should be structure as follow:
  • Cat_Dog_data/train/cat, Cat_Dog_data/train/dog for the training dataset
  • Cat_Dog_data/test/cat, Cat_Dog_data/test/dog for the test dataset
  1. cat_dog_tranfer_learning content the code of the classicication of Dog vs Cat using a pretrained model
  2. Dog_Cat_Classifier content the code for the classicication with 2 and 4 convolutions

Note

As bonus to this repository I also build a face landmark of the student of AMMI 2019

About

Dogs and Cats Classification with Convolutional Neural Network (CNN)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors