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
498 stars 139 forks source link

StopIteration #21

Open gydlcc opened 6 years ago

gydlcc commented 6 years ago

stn-ocr stn-ocr

Hello dear B: it is seems like only run 10 epochs how could i resolve this ?!

Bartzi commented 6 years ago

Yes running for only 10 epochs is intended behaviour. You can set the number of epochs with the command-line switch --num-epochs.

gydlcc commented 6 years ago

yeah, thank you, as in the picture i uploaded, i wonder that the information between Traceback and StopIteration, is that t means i can't plot the logs? i wanna plot logs, how could i solve it.

Bartzi commented 6 years ago

I think it can not plot the logs, because the log output of MXNet is not as expected (this changes from time to time, especially if you are using a version of MXNet, other than 0.9.3). In order to fix this you would have to check the log parsing code in utils/plot_log.py and determine whats different with the log_output and fix the code :sweat_smile:

gydlcc commented 6 years ago

alright, thank you pretty much

gydlcc commented 6 years ago

Hello there, sorry to bother you again. I noticed that your groundtruth file only include label and didn't include the groundtruth bbox coordinates, if my own data have groundtruth bbox coordinates, is that code still work?

Bartzi commented 6 years ago

I think it should still work, but you will need to separate the bbox coordinates from the labels.

gydlcc commented 6 years ago

Alright, thank you pretty much

avasisht-celadon commented 5 years ago

hi, I still get the above error with 100 epochs. Could you please tell how I would solve it? capture

Bartzi commented 5 years ago

It seems that the output of your training looks different than what is expected by the log_plotter are you sure that you are using MXNet in version 0.9.3?

avasisht-celadon commented 5 years ago

Yes, it was not v0.9.3. I cloned mxnet again and checked out v0.9.3. Then, I issued "make" to build mxnet. But, I do get the error:-

Makefile:27: mshadow/make/mshadow.mk: No such file or directory Makefile:28: /home/mxnet/dmlc-core/make/dmlc.mk: No such file or directory Makefile:126: /home/mxnet/ps-lite/make/ps.mk: No such file or directory make: *** No rule to make target '/home/mxnet/ps-lite/make/ps.mk'. Stop.

I then did git clone --recursive , the same above error persists.

avasisht-celadon commented 5 years ago

I solved the above issue, by cloning the submodules. But, Now, ended up with lot of compilation errors. These errors are absent in latest mxnet versions though.

Bartzi commented 5 years ago

Well, that is a problem... You could try to use a more modern version of MXNet, but then you'll need to adapt the code that examines the log file and scrapes all the number necessary for plotting from it. You could PR your changes that make the code compatible to a newer version if you like ;)