Skip to content

VictoriaAlbanese/Cat-vs-Dog-Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

50 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Cat-vs-Dog-Classifier

This repository contains our Computer Vision final project where we classify images of cats and dogs utilizing Tensorflow and Scikit-Image. Project created by Victoria Albanese and Hannah Chiodo (2019)

Dataset Source: https://www.kaggle.com/c/dogs-vs-cats/data

Paper: https://docs.google.com/document/d/1FdlxyWHFFI08Zrb3GElL8cCfKouTdnqnseJMfkwAaSU/edit?usp=sharing

Presentation Slides: https://docs.google.com/presentation/d/1bKwMss2SqnEfuqEztXtnBpc3USlz1AN5UPHB_fzIUXc/edit?usp=sharing

Dataset directory structure is "{project location}\dataset\test_data" and "{project location}\dataset\training_data"

Python version: 3.6.4

Required packages:
-keras
-sklearn
-skimage

Code execution:
1) Split dataset manually: test_data should contain cats and dogs 0-3124. The rest go in training_data
2) Creating the models: run nn_functions.py for the desired extractor (i.e. EXTRACTOR = fe.HOG_extractor). For the HOG, set PIXELS_PER_CELL to the desired value, i.e. "4x4"
3) Testing the models: run test_nn.py for the desired model. The model type is the EXTRACTOR value set in nn_functions.py

We used this tutorial as a jumping off point for our neural net and data processing: https://www.pyimagesearch.com/2016/09/26/a-simple-neural-network-with-python-and-keras/

   ,';,               ,';,
 ,' , :;             ; ,,.;
 | |:; :;           ; ;:|.|
 | |::; ';,,,,,,,,,'  ;:|.|    ,,,;;;;;;;;,,,
 ; |''  ___      ___   ';.;,,''             ''';,,,
 ',:   /   \    /   \    .;.                      '';,
 ;    /    |    |    \     ;,                        ';,
;    |    /|    |\    |    :|                          ';,
|    |    \|    |/    |    :|     ,,,,,,,               ';,
|     \____| __ |____/     :;  ,''                        ;,
;           /  \          :; ,'                           :;
 ',        `----'        :; |'                            :|
   ',,  `----------'  ..;',|'                             :|
  ,'  ',,,,,,,,,,,;;;;''  |'                              :;
,'  ,,,,                  |,                              :;
| ,'   :;, ,,''''''''''   '|.   ...........                ';,
;       :;|               ,,';;;''''''                      ';,
 ',,,,,;;;|.............,'                          ....      ;,
           ''''''''''''|        .............;;;;;;;''''',    ':;
                       |;;;;;;;;'''''''''''''             ;    :|
                                                      ,,,'     :;
                                          ,,,,,,,,,,''       .;'
                                         |              .;;;;'
                                         ';;;;;;;;;;;;;;'

About

This repository contains our Computer Vision final project where we classify images of cats and dogs utilizing Tensorflow and Scikit-Image.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages