Skip to content

Bartz36/Surfboard_VS_Boogieboard_Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Surfboard vs Boogieboard Image Classifier 🏄‍♂️🏖️

A deep learning image classifier trained to distinguish between surfboards and boogie boards using TensorFlow and Keras.


Project Overview

This project uses a convolutional neural network (CNN) trained on a custom dataset of surfboards and boogie boards. The model takes in an image and predicts whether it's a boogie board or surfboard, along with the prediction confidence.


Technologies Used

  • Python
  • TensorFlow / Keras
  • NumPy, Pillow
  • Anaconda & VS Code

Project Structure

SurfVsBoogieClassifier/
├── train.py         # Script to train the CNN model
├── predict.py       # Script to predict a new image
├── model.h5         # Trained model (included for testing)
├── data/            # Contains 'train' and 'test' image folders
│   ├── train/
│   │   ├── boogies/
│   │   └── surfboards/
│   └── test/
│       ├── boogies/
│       └── surfboards/
├── .gitignore       # Ignores model and cache files
└── README.md        # Project overview and usage

About

Machine Learning classifier which can differentiate between surfboards and boogieboards

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages