Bartzi / stn-ocr

Code for the paper STN-OCR: A single Neural Network for Text Detection and Text Recognition
https://arxiv.org/abs/1707.08831
GNU General Public License v3.0
499 stars 137 forks source link

Evaluation fail! #31

Open ThaiLe189 opened 4 years ago

ThaiLe189 commented 4 years ago

Hello Bartzi I try to run evaluation SVHN and i get this error: Screenshot from 2020-08-17 13-49-29

the command: python eval_svhn_model.py '/home/hthai/stn-ocr/datasets/svhn/original_svhn/models/model' 0040 '/home/hthai/stn-ocr/datasets/svhn/evaluation/test.csv' '/home/hthai/stn-ocr/datasets/svhn/svhn_char_map.json'

How can i fix it ? Thank you !

Bartzi commented 4 years ago

Sorry that it took me so long to answer!

That is definitely difficult to say. What version of MXNet are you using? Did you make any changes in the code?

ThaiLe189 commented 4 years ago

Yeah i using another version of mxnet because when i compile mxnet with tag v0.9.3, i get error

The command i was ran: git clone --recursive https://github.com/apache/incubator-mxnet.git mxnet1 git checkout v0.9.3

edit config file in mxnet

Screenshot from 2020-08-27 13-44-13

git submodule update --init --recursive make -j4 \USE_OPENCV=1 \USE_CUDA=1 \USE_CUDA_PATH=/usr/local/cuda \USE_CUDNN=1 \USE_BLAS=openblas

Screenshot from 2020-08-27 13-50-24

in image is mxnet1 because i don't want to remove version of mxnet i was compiled

ThaiLe189 commented 4 years ago

I realize that compile mxnet with cuda give me that error, i have try compile mxnet with command: make -j4 \USE_OPENCV=1 \USE_BLAS=openblas and it work but i can't use my gpu for train I using Cuda version 10.1 and cudnn 7.6.5

Update i try use Cuda 9.0 but have same error

Bartzi commented 4 years ago

Hmm, I think I can not help with that then, since I only tested the code with the version of MXNet showed in the README. I suggest that you have a look at my other implementation with Chainer from the paper SEE. This should work better on newer systems, since chainer is a nicer framework.

Bartzi commented 4 years ago

Oh and regarding your last screenshot: What is the actual error? I can only see warnings and that there is an error, while compiling cuddn_convolution

ThaiLe189 commented 4 years ago

I compile mxnet v0.9.3 with command: make -j4 \USE_OPENCV=1 \USE_CUDA=1 \USE_CUDA_PATH=/usr/local/cuda \USE_CUDNN=1 \USE_BLAS=openblas and i get that error, i don't know actual error So let me check Chainer first Thank you very much.