7-Segment-Digit-OCR

Red 7 segment digit OCR.

---

Install

git clone https://github.com/tdat97/7-Segment-Digit-OCR.git
cd 7-Segment-Digit-OCR
pip install -r requirments.txt

---

Example

from utils.ocr import OcrEngine
import matplotlib.pyplot as plt

img = plt.imread("./temp/test_digit.jpg")
plt.imshow(img)

ocr_engine = OcrEngine("./models/7seg_ocr.h5")
print(ocr_engine(img)) # '267'

---

Loss Figure

---

Prediction Plot