Skip to content

This project was carried out using OpenCV Python and libraries such as numpy, matplotlib and scipy. It aims at classifying the images into three categories, namely, airplanes, cars and motorcycles using an Artificial Neural Network. Similar to humans, machines, too, require to analyze the features of an image to determine its content. Hence, for…

Notifications You must be signed in to change notification settings

wiryawan46/Natural-Image-Classification

Repository files navigation

Natural-Image-Classification

Run ANN_train.py to classify the images using HoG and SURF features and run ANN_train_daubechies.py to classify the images using Daubechied D4 Wavelet Transform features. Use the following terminal command to run both the codes:

"python -W ignore ANN_train.py"

OR

"python -W ignore ANN_train_daubechies.py"

(Note: The code "ANN_train.py" gives a path to the database I have used for this project. You will need to give the database path as it is on your system).

Both the codes first train the back propagation network using the training data and later test it with the images not used in the training dataset. Since, we are using a total of 480 images as our database (with 160 images belonging to each category), the code will take approximately 10-12 minutes to display the classification results (i.e. the annotated test image).

We could have taken the entire dataset (i.e. a total of 2300 images) for training and testing purposes, however, that would have taken a huge amount of running time for the codes. Hence, for display purposes, we have reduced the size of our dataset to a total of 480 images.

About

This project was carried out using OpenCV Python and libraries such as numpy, matplotlib and scipy. It aims at classifying the images into three categories, namely, airplanes, cars and motorcycles using an Artificial Neural Network. Similar to humans, machines, too, require to analyze the features of an image to determine its content. Hence, for…

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages