Skip to content

nesi73/CoinDetector

Repository files navigation

CoinDetector

Detector de monedas mediante visión por computador, para poder usar el dector de monedas con yolovs5 utilizar el fichero de generate_database.py, con el cual se creará una base de datos con todas las imagenes, divididas tanto en entrenamiento, validación y test, por último ejecutar coins_detector2.py para detectar aquellas imágenes que tengan monedas y consiga poner las etiquetas normalizadas para yolo necesarias.

generate_dataset.py

Con este script conseguiremos ampliar el tamaño de nuestro dataset y dividirlo en diferentes carpetas, siendo estas train, val y test. También se crearán 3 csv diferentes uno para cada conjunto de datos.

coin_detector2.py

Se tiene que realizar tanto para train, val y test el siguiente comando:

python .\coins_detector2.py --folder test

Lo que conseguiremos con esto será la generación de una carpeta label por cada carpeta train, val y test. Estas labels se encuentran normalizadas y en el formato adecuado para poder entrenar posteriormente YOLO, las labels se componen de la clase correspondiente, las coordenadas x e y normalizadas del centro de la moneda, y el tamaño y la altura también normalizadas.


Con esto conseguiremos una distribución como la siguiente:


├───detected_coins
│ ├───test
│ │ ├───images
│ │ └───labels
│ ├───train
│ │ ├───images
│ │ └───labels
│ └───val
│ ├───images
│ └───labels

coin_detector.py

Este script sirve para detectar monedas, valdría con poner de manera paralela la cámara enfrente de las monedas posicionadas delante de un papel blanco, posteriormente al correr el script se remarcarán las monedas con su correspondiente valor, por la terminal se podrá observar el valor total de las monedas

About

Coin detector with computer vision

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages