Sampahku

image-classification on type of waste
Demo Sampahku app click here

Depedencies

• tensorflow 2.9.1
• python 3.10

Dataset

For the dataset we use These Github dataset from Sarah Frost, Bryan Tor, Rakshit Agrawal, Angus G. Forbes use for compostnet. The original dataset contain 7 class of waste type glass, paper, cardboard, plastic, metal, trash with contain 2.751 photo of waste in total. But for our dataset we deleted the trash class so the final class that we use for this project is:

• cardboard
• compost
• glass
• metal
• paper
• plastic

Architecture

For our model we use transfer learning method for this project, We use VGG16 architecture which is a convolutional neural network that is 16 layers deep and we add a few more layers on top of that, you can see the detail about our model below

Result

accuracy and validation accuracy

we achieve pretty good number consistenly 90% training accuracy with 80% accuracy on validation

accuracy loss and validation loss

our losses also is preety low