Bugiene Machine Learning Model
This repository contains the code for a machine learning model that performs implied computer vision tasks using a Convolutional Neural Network (CNN). The machine learning model was developed to classify fruits into rotten or fresh categories using computer vision and a convolutional neural network (CNN). The model was built using the TensorFlow Keras library. In this case, the CNN model was trained on a dataset of images of fruits, such as: apples, avocados, bananas, grapes, guavas, oranges. The dataset included images of both rotten and fresh fruits. The CNN was able to learn to distinguish between the two types of fruit based on the features it extracted from the images.The output from our model is prediction result "Fresh" or "Rotten" and the accuracy.
Table of Contents
Dataset
The model is trained on a custom dataset consisting of labeled images. The dataset can be obtained from https://www.kaggle.com/datasets/sriramr/fruits-fresh-and-rotten-for-classification and https://www.kaggle.com/datasets/moltean/fruits.
Model Architecture
The CNN model architecture used for this project is as follows:
pre_trained_model = VGG16(input_shape=(150,150,3), include_top=False) for layer in pre_trained_model.layers: layer.trainable = False x = layers.Dense(1024, activation='relu')(x) x = layers.Dropout(0.2)(x) x = layers.Dense(1, activation='sigmoid')(x)
Requirements
To run the code in this repository, you will need the following dependencies:
Python [3.10.10]
TensorFlow [2.12.0]
Usage
Clone this repository to your local machine.
Install the required dependencies by running
pip install -r requirements.txt
.Clone this https://github.com/Bugiene/Bugiene-app/blob/master/machine-learning/bugiene_model.ipynb to your local machine.
Run the main script using
python main.py
.
Results
Found 4004 images belonging to 2 classes. 4004/4004 [==============================] - 41s 10ms/step - loss: 0.1405 - accuracy: 0.9451 accuracy test: 0.9450549483299255 loss test: 0.14052222669124603
And for the result test you can check this https://github.com/Bugiene/Bugiene-app/tree/master/machine-learning/result-test
Contributor Acknowledgment
We would like to acknowledge the following contributors for their valuable contributions to this project:
- Deni Irawan (GitHub: Densu341)
- Sandro Sinaga (GitHub: SandroSinaga24)
- Laila Nur Anggamurti (GitHub: jejukyul)
Contact
For any questions or inquiries, please contact the contributor mentioned above. Thank you.