search

jomjol / AI-on-the-edge-device

Easy to use device for connecting "old" measuring units (water, power, gas, ...) to the digital world
https://jomjol.github.io/AI-on-the-edge-device-docs/
5.67k stars 612 forks source link

Neural Network: zero or six? #107

Closed benjholl closed 3 years ago

benjholl commented 3 years ago

Thanks alot for making this great project happen, i really love it and have two units up and running! Unfortunately, on my watermeter the zero gets interpreted as 6 quite often. not always, but i would guess around 50% of the time it will output "6" instead of "0"

regarding to the neural network this is a zero: 0_ROI1_20210207-101635 and this is a six: 6_ROI1_20210207-120635

Is there a instruction on how someone can train the network (or maybe even create a new network) for this system?

jomjol commented 3 years ago

Hallo @benjholl,

the instructions can be found in this repository: https://github.com/jomjol/neural-network-digital-counter-readout

I update the tflite-files from time to time. So if yous send me some representatives of your digits (about 2-3 per digit), I will incorporate them in the next training. This can be done quite fast, as soon as I have the complete set of images.

Best regards, jomjol