jayzstep / tiralabra

0 stars 0 forks source link

Tiralabra

NumNet on neuroverkko, joka tunnistaa käsin piirrettyjä numeroita MNIST-tietokannan datasta noin 97% tarkkuudella.

Mukana myös localhostiin käynnistyvä GUI, jossa voi klikata samplekuvia verkolle tunnistettavaksi.

Voit itse valita koulutatko verkon uudestaan, vai käytätkö valmiiksi koulutettua verkkoa komennoilla "train" tai "run". Tarkemmat ohjeet alla. Koulutuksessa menee 1-10 minuuttia, riippuu koneen spekseistä.

Lyhyesti: Lataa repo, lataa koulutusdata ja samplet, anna Poetryn laulaa, fool around.

Käyttöohje

Kopioi repositio

git clone https://github.com/jayzstep/tiralabra.git
cd NumNet
mkdir data

Lataa koulutusdata, eli MNIST-tietokanta

Lataa täältä molemmat CSV-tiedostot sekä samplekuvat ja laita/pura ne tässä muodossa kansioon NumNet/data/

Asenna riippuvuudet (tarvitset Poetryn)

poetry install

Käynnistä shell

poetry shell

Treenaa verkko (jos haluat)

cd numnet
python3 main.py train

Tämä myös käynnistää GUI:n localhostiin ja näyttää komean plotin koulutuksen kulusta. Seuraa ruudun ohjeita.

TAI

Käynnistä äppi suoraan

Treenivaiheen voi skipata ja serverin käynnistää numnet/ -kansiosta suoraan komennolla:

python3 main.py run

GUI käynnistyy localhostiin, seuraa ruudun ohjeita.

Testaus

Testit voi suorittaa NumNet/ -kansiosta käsin komennolla:

pytest