{"id":23103882,"url":"https://github.com/moatifbutt/waste-classification","last_synced_at":"2025-04-03T21:44:43.782Z","repository":{"id":247245779,"uuid":"337083508","full_name":"moatifbutt/waste-classification","owner":"moatifbutt","description":null,"archived":false,"fork":false,"pushed_at":"2021-02-08T17:17:52.000Z","size":1726,"stargazers_count":0,"open_issues_count":0,"forks_count":1,"subscribers_count":1,"default_branch":"main","last_synced_at":"2025-02-09T09:43:58.390Z","etag":null,"topics":[],"latest_commit_sha":null,"homepage":null,"language":"Jupyter Notebook","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":null,"status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/moatifbutt.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":null,"code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2021-02-08T13:23:17.000Z","updated_at":"2021-02-08T17:17:54.000Z","dependencies_parsed_at":"2024-07-07T16:28:46.232Z","dependency_job_id":"5fde2bed-0f25-49da-9ac0-b8df73f6e437","html_url":"https://github.com/moatifbutt/waste-classification","commit_stats":null,"previous_names":["moatifbutt/waste-classification"],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/moatifbutt%2Fwaste-classification","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/moatifbutt%2Fwaste-classification/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/moatifbutt%2Fwaste-classification/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/moatifbutt%2Fwaste-classification/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/moatifbutt","download_url":"https://codeload.github.com/moatifbutt/waste-classification/tar.gz/refs/heads/main","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":247085993,"owners_count":20881158,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":[],"created_at":"2024-12-17T00:29:39.474Z","updated_at":"2025-04-03T21:44:43.756Z","avatar_url":"https://github.com/moatifbutt.png","language":"Jupyter Notebook","funding_links":[],"categories":[],"sub_categories":[],"readme":"# Waste Classification Using Convolutional Neural Network\n\n## Pre-Requisites\n1. Anaconda (https://www.anaconda.com/products/individual)\n2. Pytorch (https://pytorch.org/get-started/locally/)\n3. Waste Classification Binary Class Dataset (https://www.kaggle.com/techsash/waste-classification-data)\n4. Thung and Wang's Dataset (https://github.com/garythung/trashnet/blob/master/data/dataset-resized.zip)\n\n## Data Augmentation\n\nTo add more diversity in dataset, we have performed three types of data augmentations including,\n\n- Gaussian Blur\n- Horizontal Flip\n- Contrast\n\nYou can use the **Augmentation.ipynb** in **Train_Notebooks** to perform augmentation on the provided dataset.\n\n## Training\n\nTo this end, three state-of-the-art CNNs i.e. **GoogleNet** , **Resnet-50**, **VGG**, and **ResNext** are employed to perform classification. Based on the best performed, we selected **ResNext**, and added a new classification block has been added in the basis of network. It is important to mention here that, pre-trained CNNs are downloaded, and trained through transfer learning to achieve effective results. All the networks are available in above-folder named **Trained Notebooks**. All the trained models are available in above-folder **models**.\n\n## Results\nThe experiments are performed on both i.e. the augmented and non augmented datasets to analyse the difference through adaptive and static learning rate. The experiments showed that the modified ResNext achieved better accuracy on augmented dataset with adaptive learning rate.\n\n#### GoogleNet\n- With Adaptive Learning Rate on non-Augmented Dataset\n\n![alt text](https://github.com/aatiibutt/waste-classification/blob/main/graph/google-non.PNG?raw=true)\n\n- With Adaptive Learning Rate on Augmented Dataset\n\n![alt text](https://github.com/aatiibutt/waste-classification/blob/main/graph/google-aug.PNG?raw=true)\n\n#### Resnet\n\n- With Adaptive Learning Rate on non-Augmented Dataset\n\n![alt text](https://github.com/aatiibutt/waste-classification/blob/main/graph/res-non.PNG?raw=true)\n\n- With Adaptive Learning Rate on Augmented Dataset\n\n![alt text](https://github.com/aatiibutt/waste-classification/blob/main/graph/res-aug.PNG?raw=true)\n\n#### VGG\n\n- With Adaptive Learning Rate on non-Augmented Dataset\n\n![alt text](https://github.com/aatiibutt/waste-classification/blob/main/graph/vgg-non.PNG?raw=true)\n\n- With Adaptive Learning Rate on Augmented Dataset\n\n![alt text](https://github.com/aatiibutt/waste-classification/blob/main/graph/vgg-aug.PNG?raw=true)\n\n#### ResNext\n\n- With Adaptive Learning Rate on non-Augmented Dataset\n\n![alt text](https://github.com/aatiibutt/waste-classification/blob/main/graph/resnxt-non.PNG?raw=true)\n\n- With Adaptive Learning Rate on Augmented Dataset\n\n![alt text](https://github.com/aatiibutt/waste-classification/blob/main/graph/resnxt-aug.PNG?raw=true)\n\n#### ResNext-Plus\n\n- With Adaptive Learning Rate on non-Augmented Dataset\n\n![alt text](https://github.com/aatiibutt/waste-classification/blob/main/graph/resnxt-non.PNG?raw=true)\n\n- With Adaptive Learning Rate on Augmented Dataset\n\n![alt text](https://github.com/aatiibutt/waste-classification/blob/main/graph/respls-aug.PNG?raw=true)\n\n#### ResNext-Plus on Sencond Dataset (Thung and Yang’s)\n\n![alt text](https://github.com/aatiibutt/waste-classification/blob/main/graph/resnxtpls-seconddata.PNG?raw=true)\n\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fmoatifbutt%2Fwaste-classification","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fmoatifbutt%2Fwaste-classification","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fmoatifbutt%2Fwaste-classification/lists"}