/Solid-waste-classification-in-Real-time

Classifying solid waste into plastics, metal, paper etc. in real time.

Primary LanguageJupyter Notebook

Solid-waste-classification-in-Real-time

Download dataset from here_

  1. 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'.

  2. Classifier.ipynb - Accuracy comparision between RandomForest, XGBoost and DecisionTree. RandomForest parameter tuning on hist.csv data and saving trained model for future predictions.

  3. RealTimePredictions.ipynb - Classifying solid wastes into one of the five categories in real time.