Ecosyste.ms: Awesome

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

Awesome Lists | Featured Topics | Projects

https://github.com/lasithaamarasinghe/dog-classification-with-tensorflow

This is an end-to-end deep learning project using Tensorflow and Keras for dog classification.
https://github.com/lasithaamarasinghe/dog-classification-with-tensorflow

convolutional-neural-networks deep-learning dog-classification jupyter-notebook keras matpolotlib pandas pil pillow tensorflow

Last synced: about 2 months ago
JSON representation

This is an end-to-end deep learning project using Tensorflow and Keras for dog classification.

Awesome Lists containing this project

README

        

# Dog-Classification-with-Tensorflow

![image](https://github.com/LasithaAmarasinghe/Dog-Classification-with-Tensorflow/assets/106037441/a857021a-c305-4d4f-8a31-7ea70a54e985)

## Overview

* This is an end-to-end deep learning project using [**Tensorflow**](https://www.tensorflow.org/) and [**Keras**](https://keras.io/).
* A dataset of dog images is read, and then a convolutional neural network is trained to classify the dogs by breed.
* This repository contains all the codes and resources used to build this dog classification system.

## Steps

* Loading in data with TensorFlow and Keras
* Exploring images in data with PIL (Python Imaging Library)
* Training a neural network with TensorFlow and Keras
* Looking at model error
* Improving model accuracy with dropout
* Augmenting data with Keras
* Investigating model error

## Code
You can find the code for this project here.
* [dog classification.ipynb](https://github.com/LasithaAmarasinghe/Dog-Classification-with-Tensorflow-/blob/main/dog%20classification.ipynb)

## Technologies/Tools

* JupyerLab / [Google Colab](https://colab.research.google.com/)
* Python 3.10.12
* Tensorflow - `pip install tensorflow`
* Python packages
* Pandas - `pip install pandas`
* Pillow - `pip install pillow`
* Matplotlib - `pip install matplotlib`

![Python](https://img.shields.io/badge/python-3670A0?logo=python&logoColor=FFFF00)
![Jupyter Notebook](https://img.shields.io/badge/jupyter-%23FA0F00.svg?logo=jupyter&logoColor=white)
![TensorFlow](https://img.shields.io/badge/TensorFlow-%23FF6F00.svg?logo=TensorFlow&logoColor=white)
![Keras](https://img.shields.io/badge/Keras_-black?logo=keras)
![Pandas](https://img.shields.io/badge/pandas_-%20green?logo=pandas)
![Pillow](https://img.shields.io/badge/pillow_-blue)
![Matplotlib](https://img.shields.io/badge/matplotlib_-purple)

## Data

Data used are from the [Stanford](http://vision.stanford.edu/aditya86/ImageNetDogs/)

You can download the dog image dataset used in this project here.

* [dog_images.zip](https://drive.google.com/uc?export=download&id=1sj62C-9WKD09-8iYSeEvXmAGQoY2oFFQ) - please unzip this file into a folder called `images`.

## Output

![image](https://github.com/LasithaAmarasinghe/Dog-Classification-with-Tensorflow-/assets/106037441/05d2f9bd-c320-4082-9a20-1d64bcb1c7ef)
![image](https://github.com/LasithaAmarasinghe/Dog-Classification-with-Tensorflow/assets/106037441/e49ec227-f5c2-4958-a5b6-8378cf56274e)