BrailleNet is a convolutional neural network (CNN) designed for the classification of Braille characters. This repository includes the code for training the model using the Braille Dataset and provides an overview of the network architecture.
- The Braille Dataset is organized in the following structure:
- Create a directory to store images for training and validation:
- Run the script
prepare_dataset.py
to organize the images into directories corresponding to their respective characters:
dataset/
└── Braille Dataset/
└── Braille Dataset/
├── a1.JPG
├── a2.JPG
├── ...
└── z9.JPG
$ mkdir ./images/
$ python prepare_dataset.py
- Install the required dependencies:
- Run the training script:
- The trained model will be saved as
BrailleNet.h5
, and training progress will be stored in thehistory
object.
$ pip install -r requirements.txt
$ python train_model.py
The CNN architecture includes separable convolutional layers, batch normalization, and dropout for robust feature extraction. The model is trained on Braille characters with data augmentation.
- After training, the model is evaluated on a validation set, and the accuracy is printed.
- To load the trained model and evaluate it:
$ python evaluate_model.py
- Python 3.x
- TensorFlow
- Keras
- Matplotlib (for visualization)
Feel free to customize the model architecture or experiment with hyperparameters based on specific requirements.
This project is licensed under the MIT License - see the LICENSE file for details.