YOLOv4 for Water Meter Reading

The goal of the project is to read the exact consumption of water in cubic meters. As shown on the picture below, the cubic meters are written in white on a black background. As YOLO is one of the best Convolutional Neural Network (CNN) algorithms for object detection, we decided to implement our model with a Pytorch version of YOLOv4. We defined 12 classes of objects that we want to detect on the pictures: the 10 digits from 0 to 9, the part of the meter corresponding to the liters, and the whole counter. We labelled each picture by drawing the bounding boxes of each object found.