hello2all / GTSRB_Keras_STN

German Traffic Sign Recognition Benchmark, Keras implementation with Spatial Transformer Networks
MIT License
49 stars 15 forks source link

Accuracy is significantly lower than 99.5% #4

Closed tabVersion closed 5 years ago

tabVersion commented 5 years ago

Hello Please check you uploaded the correct weight file image

hello2all commented 5 years ago

Thank you for your interest in this project. The uploaded weight file is correct and has been verified many times. You may use the provided test_model.ipynb notebook to view the testing accuracy. However, it has been quite a long time since completetion of this project and the newer versions of Tensorflow and Keras might have compatibility issues with older weight files. Could you:

  1. Run the notebook and see if the testing accuracy is correct
  2. If not, please let me know which version of Tensorflow and Keras you have been using.

The lastest correct results was tested on: Tensorflow 1.3.0 Keras 2.0.8

tabVersion commented 5 years ago

Sorry for not checking the message box. I used a wrong way to test the results. Your model is really awesome. I retrained the model and get 99% accuracy after 130 epochs.

sisrfeng commented 3 years ago

Hi, what versions of these packages should we use?

python
matplotlib
scipy
numpy
pandas
scikit-learn

thx!