This repository contains a deep learning-based solution for classifying maggots into various growth stages using TensorFlow and Keras. The application aims to assist maggot farmers in identifying and categorizing maggots based on images captured through a camera.
- Multi-Class Classification: Identify six maggot growth stages:
larva_tahap_1larva_tahap_2larva_tahap_3maggotprapupapupa
- Image Augmentation: Enhanced generalization with data augmentation.
- Pre-Trained Model Support: Compatible with transfer learning for higher accuracy.
The dataset consists of labeled images for each maggot growth stage.
Click the button below to access the dataset folder on Google Drive:
Follow the steps below to use the notebook and replicate the model training process.
git clone https://github.com/your-username/maggot-classification.git
cd maggot-classificationInstall the required Python packages:
pip install -r requirements.txtClick the button above to download the dataset. Extract the contents and place them in the data/ directory as follows:
maggot-classification/
├── data/
│ ├── train/
│ │ ├── larva_tahap_1/
│ │ ├── larva_tahap_2/
│ │ ├── larva_tahap_3/
│ │ ├── maggot/
│ │ ├── prapupa/
│ │ ├── pupa/
│ ├── validation/
│ ├── larva_tahap_1/
│ ├── larva_tahap_2/
│ ├── larva_tahap_3/
│ ├── maggot/
│ ├── prapupa/
│ ├── pupa/
Open the Jupyter Notebook maggot_classification.ipynb Follow the steps in the notebook to:
- Load and preprocess the dataset.
- Train the model using your dataset.
- Evaluate and save the trained model.