Deep Learning Classification problem using Convolutional Neural Network.
This Github links helps in classifying image as a mountain bike or a road bike. Add data to data folder divided in mountain bike and road bikes for training
The Python 3 Anaconda Distribution is the easiest way to get going with the notebooks and code presented here.
(Optional) You may want to create a virtual environment for this repository:
conda create -n cv python=3
source activate cv
This repository requires the installation of a few extra packages, you can install them all at once with:
pip install -r requirements.txt
python train.py
All Images should be jpg format
This should be used for evaluating images for testing. So the images should be classified to calculate the testing accuracy
python test.py
python test.py -i file_path.jpg
Returns an ouput whether its a road bike or a mountain bike