Trash sorter basico (Clasificador de residuos) creado en Python usando IA, redes neuronales convolucionales y machine learning para entrenarse ella misma y clasificar diferentes tipos de residuos siguiendo el dataset de Kaggle: Garbage Classification 12 categorias (ir al link para mas referencia)
Antes de usar el proyecto, tendra que instalar las dependencias almacenadas en el archivo: requeriments.txt
Para instalar el archivo, escribe esto en el terminal:
pip install requeriments.txt❗IMPORTANTE❗
El repositorio ya tiene un modelo pre-entrenado: waste_classifier_model.h5 sin embargo, en caso de que el archivo se pierda o se quiera recompilar con otros valores, se tiene que realizar lo siguiente:
Escribir en el terminal:
python run src\full_model.py (dependiendo del dataset y los valores predefinidos el compilador puede tardarse)
Una vez con el modelo en el directorio base, se tiene ejecutar en el terminal el siguiente comando:
python run src\waste_app.py NT: esta app usa la libreria Streamlit, diseñada para crear UIs simples en Python.
Tipo de desecho: clothes/ropa
Estadisticas usando modelo preentrenado:
Tipo de desecho: cardboard/carton
Estadisticas usando modelo preentrenado:
Tipo de desecho: battery/bateria
Estadisticas usando modelo preentrenado:
NT: Hay tipos de desechos con prediccion mas favorables que otros, si se quiere ser mas exacto en clases especificas, es necesario usar otros datasets.
El archivo waste_classifier_model.h5 tiene que estar en la ruta base del proyecto, de lo contrario, saldra un mensaje indicando que no hay un modelo existente.