A deep learning image classifier trained to distinguish between surfboards and boogie boards using TensorFlow and Keras.
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.
- Python
- TensorFlow / Keras
- NumPy, Pillow
- Anaconda & VS Code
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