Skip to content

Latest commit

 

History

History
28 lines (22 loc) · 1.22 KB

README.md

File metadata and controls

28 lines (22 loc) · 1.22 KB

Cat-vs-Dog-Classification

use CNN / DNN to classify the label of the image (cat/dog); image argumentation / transfer learning -> 95% accuracy or above

Configuration

  • tensorflow == 2.7 & python == 3.8.0 (very important as tensorflow > 2.7 changes preprocess api)
  • pillow (any version) for image loader
  • matplotlib, numpy for data processing and graph

Setup

Manually

Linux

Transfer learning

Tips

  • Don't recommand run it locally!!!
  • Otherwise run it applying GPUs:

pip install tensorflow-gpu
import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))