Duplicate from hardarakha/trash-classification

Co-authored-by: Hardatama Rakha Ugraha <hardarakha@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,60 @@
+---
+datasets:
+- garythung/trashnet
+---
+# Trash Classification Model
+
+This repository contains code for training a Convolutional Neural Network (CNN) model to classify images of trash. The model is trained on the **TrashNet** dataset, which is available through the Hugging Face datasets library.
+
+## Project Overview
+
+The goal of this project is to build a machine learning model that can classify images of trash into categories such as plastic, metal, paper, etc. This model could potentially be used in applications to aid in waste management and recycling efforts.
+
+## Dataset
+
+The dataset used for this project is **TrashNet**, a labeled collection of images of different types of trash. I load this dataset using the [Hugging Face Datasets library](https://huggingface.co/docs/datasets).
+The dataset includes the following classes:
+- Plastic
+- Metal
+- Paper
+- Cardboard
+- Glass
+- Trash (other miscellaneous items)
+
+## Prerequisites
+
+- Python 3.7 or higher
+- Git
+- [Hugging Face Datasets](https://github.com/huggingface/datasets), TensorFlow, and other dependencies listed in `requirements.txt`
+
+## Github Repository
+
+For the detailed description about the training process and models' performace, you can visit this [Github Repository](https://github.com/ardrkha/trash-classification)
+
+## Usage
+1. Clone the repository
+```
+git clone https://github.com/ardrkha/trash-classification.git
+cd trash-classification
+```
+
+2. Run the notebook
+```
+trash-classification/
+├── notebook/
+│   ├── [Notebook.ipynb] (run this notebook for the complete implementation)
+├── README.md
+└── ...
+```
+
+## License
+
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
+
+## Contact
+
+If you have any questions, feel free to reach out to us at hardatama27@gmail.com.
+
+---
+
+Developed by Hardatama Rakha Ugraha - 2024

trash_classification_model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:56f2c9e8cb148359f2711e2fb611f0ea42a001b764475b9ee8301279ae043d00
+size 12594048