KNN-Math

Tento projekt slouží k rozpoznávání matematických výrazů z obrázků a jejich převodu do LaTeX zápisu. Pro strojové učení využíváme knihovnu PyTorch.

Dokumentace

Podrobná dokumentace je dostupná ve složce dokumentace. Je psána v LaTeXu a zdrojové .tex soubory jsou k dispozici.

Příprava datasetu

Ve složce app/neural_network se nachází hlavní soubory, které definují přípravu datasetu, HME recognition model a trénovací skript. Přípravu datasestu lze sputit příkazem:

python -m app.neural_network.prepare_dataset --input .\dataset\train_set\ --output dataset_path/

Trénování lze spustit pomocí:

python -m app.neural_network.train_nn --device cuda --data_dir dataset_path/ --epochs 30

Relevantní zdroje a soutěže

• CROHME 2023 - Competition on Recognition of Online Handwritten Mathematical Expressions
• ICDAR - International Conference on Document Analysis and Recognition
• Research paper 1
• Research paper 2