https://github.com/kiritoind/neural-network-for-cat-image-classification

This script defines essential functions for constructing a deep neural network from scratch for image classification. It includes activation functions like sigmoid and ReLU, data loading from H5 files, parameter initialization, forward and backward propagation for training, gradient descent for parameter updates, and functions for prediction
https://github.com/kiritoind/neural-network-for-cat-image-classification

classification computer-vision deep-learning deep-neural-networks jupyter-notebook numpy python

Last synced: 2 days ago
JSON representation

This script defines essential functions for constructing a deep neural network from scratch for image classification. It includes activation functions like sigmoid and ReLU, data loading from H5 files, parameter initialization, forward and backward propagation for training, gradient descent for parameter updates, and functions for prediction

• Host: GitHub
• URL: https://github.com/kiritoind/neural-network-for-cat-image-classification
• Owner: kiritoInd
• Created: 2023-11-03T20:24:59.000Z (about 1 year ago)
• Default Branch: main
• Last Pushed: 2023-11-03T20:31:23.000Z (about 1 year ago)
• Last Synced: 2024-11-09T23:28:26.278Z (about 2 months ago)
• Topics: classification, computer-vision, deep-learning, deep-neural-networks, jupyter-notebook, numpy, python
• Language: Jupyter Notebook
• Homepage:
• Size: 3.01 MB
• Stars: 1
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Cat Image Classification Neural Network

A from scratch neural network for classifying cat images using deep learning.

## Introduction

This project is a  example of building a neural network to classify cat images. The neural network is trained on a dataset of labeled cat and non-cat images.

## Features

- Image classification as "cat" or "not cat."

- Pre-trained model for quick inference.

- Training scripts for custom dataset.