Download dataset from here_
-
Image_to_hist.ipynb - Image augmentation using opencv. Rotating images, adding noise and Calculating histograms of final Image dataset produced. Storing calculated histograms data in a dataframe, adding labels and saving final dataframe in 'hist.csv'.
-
Classifier.ipynb - Accuracy comparision between RandomForest, XGBoost and DecisionTree. RandomForest parameter tuning on hist.csv data and saving trained model for future predictions.
-
RealTimePredictions.ipynb - Classifying solid wastes into one of the five categories in real time