- This repository contains 3 models for Street View House Number classification.
- The first model is Histogram of Oriented Gradient combined with Neural Network
which is contained in the folder HOG_ANN. The script
train_hog_ann.pyis used to train this model on training set andeval_hog_ann.pyis used to evaluate this model on test set. - The second model is Convolutional Neural Network which is contained in the folder CNN. The script
train_cnn.pyis used to train this model on training set andeval_cnn.pyis used to evaluate this model on test set. - The third model is Histogram of Oriented Gradient combined with Support Vector Machine. The script
train_hog_svm.pyis used to train this model on training set andeval_hog_svm.pyis used to evaluate this model on test set. - To run these script first download the training and testing data from SVHN.
- To train the CNN model run
python train_cnn.py '--data_dir [path to data] --log_dir [path for log files]' - To train the HOG_ANN model run
python train_hog_ann.py '--data_dir [path to data] --log_dir [path for log files]' - To evaluate the CNN model run
python eval_cnn.py --model_dir [path to trained model] --data_dir [path to data] - To evaluate the HOG_ANN model run
python eval_hog_ann.py --model_dir [path to trained model] --data_dir [path to data] - To train the HOG_SVM model run
python train_hog_svm.py --data_dir [path_to_data] - To evaluate the HOG_SVM model run
python eval_hog_svm.py --model_dir [path to trained model] --data_dir [path to data]
-
Notifications
You must be signed in to change notification settings - Fork 0
trungtrinh44/SVHN
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published