Skip to content
Snippets Groups Projects
README.md 1.02 KiB
Newer Older
Jiale Song's avatar
Jiale Song committed
# CS4804 Mini-Project OCR

Jiale Song's avatar
Jiale Song committed
## Code Specifications
Jiale Song's avatar
Jiale Song committed

Jiale Song's avatar
Jiale Song committed
1. Language: Python3
2. Install third-party library 
    
    ```pip install -r requirements.txt```
Jiale Song's avatar
Jiale Song committed

Jiale Song's avatar
Jiale Song committed
3. Trainning dataset: 
    1. [English Handwritten Characters](https://www.kaggle.com/datasets/dhruvildave/english-handwritten-characters-dataset)   Size: 13.7 MB / 25.2 MB(after unzip)
    2. [Handwritten Digits and English Characters](https://www.kaggle.com/datasets/hrishabhtiwari/handwritten-digits-and-english-characters/) Size: 85 MB / 545.57 MB(after unzip)
Jiale Song's avatar
Jiale Song committed

Jiale Song's avatar
Jiale Song committed
    **Please place all unzipped dataset files under *dataset* folder**

Jiale Song's avatar
Jiale Song committed
4. Usage
Jiale Song's avatar
Jiale Song committed
    1. Train the model

        ```train-ocr-model.py [-h] -d DATASET [-m MODEL] [-p PLOT] [-s] [-t]```

        Example:

Jiale Song's avatar
Jiale Song committed
        ```python train-ocr-model.py -d dataset/dataset_mnist.csv -s -m model.keras -p result.png```
Jiale Song's avatar
Jiale Song committed

    2. Implement OCR by running model

        ```run-ocr-model.py [-h] -m MODEL -i IMAGE -s SIZE```
Jiale Song's avatar
Jiale Song committed

Jiale Song's avatar
Jiale Song committed
        Example:
Jiale Song's avatar
Jiale Song committed

Jiale Song's avatar
Jiale Song committed
        ```python run-ocr-model.py -m model.keras -i /test_cases/hand_writing/1-image.png -s 28```