emedvedev / attention-ocr

A Tensorflow model for text recognition (CNN + seq2seq with visual attention) available as a Python package and compatible with Google Cloud ML Engine.
MIT License
1.08k stars 256 forks source link

Error on train phase #181

Open mohammad228 opened 3 years ago

mohammad228 commented 3 years ago

I have some pic with text file contain annotation tag of pictures.But when i start train phase, show me this error: Traceback (most recent call last): File "/home/girizliii/.local/bin/aocr", line 8, in sys.exit(main()) File "/home/girizliii/.local/lib/python2.7/site-packages/aocr/main.py", line 257, in main num_epoch=parameters.num_epoch File "/home/girizliii/.local/lib/python2.7/site-packages/aocr/model/model.py", line 376, in train for batch in s_gen.gen(self.batch_size): File "/home/girizliii/.local/lib/python2.7/site-packages/aocr/util/data_gen.py", line 70, in gen if self.max_width and (Image.open(IO(img)).size[0] <= self.max_width): File "/usr/local/lib/python2.7/dist-packages/PIL/Image.py", line 2818, in open raise IOError("cannot identify image file %r" % (filename if filename else fp)) IOError: cannot identify image file <StringIO.StringIO instance at 0x7fce59bf9280>

how i can solve this? thanks