elmihailol / yet-another-digit-classifier

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

yet-another-digit-classifier

Классификация цифр при помощи нейросети

Требования

numpy scipy joblib keras tensorflow python3.6

Обучение нейросети

python3 train.py

Пример использования

python3 classifier.py
Расположение изображения:
images/2.jpg
0 	 1.0849699e-06
1 	 1.7289169e-07
2 	 0.99421084
3 	 0.001126814
4 	 2.621823e-06
5 	 3.0587373e-07
6 	 6.195467e-08
7 	 0.0046095573
8 	 4.613207e-05
9 	 2.4571727e-06
Число: 2
Расположение изображения:
<путь к изображению>

Требования к изображению

Цифра должна быть нарисована белым на черном фоне. Размер изображения желательно должен быть 28х28 пикселей, но возможна классификация изображений, имеющих больший размер. Для достижения наилучшего результата рекомендуется использовать квадратные изображения.

Файлы

  • images - примеры тестовых изображений
  • models/model.h5 - модель
  • train.py - скрипт обучения НС
  • classifier.py - скрипт распознавания цифр

About

License:Apache License 2.0


Languages

Language:Python 100.0%