试跑了一下test.py但遇到一个问题，请帮忙看一下 about attentionocr HOT 10 OPEN

It may be the mismatch CUDA version 10.0

from attentionocr.

您好 我也遇到类似的问题 您的问题解决了吗 @quuhua911 @zhang0jhon
Traceback (most recent call last):
File "test.py", line 121, in
test(args)
File "test.py", line 91, in test
model = TextRecognition(args.pb_path, cfg.seq_len+1)
File "test.py", line 23, in init
self.init_model()
File "test.py", line 37, in init_model
self.label_ph = self.sess.graph.get_tensor_by_name('label:0')
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/ops.py", line 3766, in get_tensor_by_name
return self.as_graph_element(name, allow_tensor=True, allow_operation=False)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/ops.py", line 3590, in as_graph_element
return self._as_graph_element_locked(obj, allow_tensor, allow_operation)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/ops.py", line 3632, in _as_graph_element_locked
"graph." % (repr(name), repr(op_name)))
KeyError: "The name 'label:0' refers to a Tensor which does not exist. The operation, 'label', does not exist in the graph."

from attentionocr.

You may use the pb model in docker image but test.py in this repository. You should load the pb model with text_recognition.py in docker image. Model trained from this repository is slightly different from the competition model in docker image.

from attentionocr.

您好，我现在在docker里运行python text_recognition.py，但是仍然有报错信息，实在打扰您了 ，不知道该如何修改 @zhang0jhon

from attentionocr.

python flaskapp.py

from attentionocr.

当我运行这个也是类似的错误 好奇怪啊 @zhang0jhon

from attentionocr.

docker与nvidia-docker版本，还有运行docker的命令？

from attentionocr.

You may use the pb model in docker image but test.py in this repository. You should load the pb model with text_recognition.py in docker image. Model trained from this repository is slightly different from the competition model in docker image.

请问能提供这个仓库的训练模型权重吗

from attentionocr.

Same problem here when not using docker:

File "/home/retrohelix/.virtualenvs/AttentionOCR/lib/python3.6/site-packages/tensorflow/python/framework/ops.py", line 3838, in _as_graph_element_locked
"graph." % (repr(name), repr(op_name)))
KeyError: "The name 'label:0' refers to a Tensor which does not exist. The operation, 'label', does not exist in the graph."

I tried using the models from the docker image:

ICDAR_0.5.pb
ICDAR_0.7.pb
text_recognition_5435.pb
text_recognition.pb

CUDA Version is:
Cuda compilation tools, release 10.0, V10.0.130
My python package versions are as follows (Python 3.6.9):

Package Version

---

absl-py 0.9.0
astor 0.8.1
cachetools 4.0.0
certifi 2019.11.28
chardet 3.0.4
cycler 0.10.0
Cython 0.29.14
gast 0.2.2
google-auth 1.10.0
google-auth-oauthlib 0.4.1
google-pasta 0.1.8
grpcio 1.26.0
h5py 2.10.0
idna 2.8
Keras-Applications 1.0.8
Keras-Preprocessing 1.1.0
kiwisolver 1.1.0
Markdown 3.1.1
matplotlib 3.1.2
msgpack 0.6.2
msgpack-numpy 0.4.4.3
numpy 1.18.0
oauthlib 3.1.0
opencv-python 4.1.2.30
opt-einsum 3.1.0
pip 19.3.1
protobuf 3.11.2
psutil 5.6.7
pyasn1 0.4.8
pyasn1-modules 0.2.7
pycocotools 2.0.0
pyparsing 2.4.6
python-dateutil 2.8.1
pyzmq 18.1.1
requests 2.22.0
requests-oauthlib 1.3.0
rsa 4.0
setuptools 42.0.2
six 1.13.0
tabulate 0.8.6
tensorboard 1.14.0
tensorflow 1.14.0
tensorflow-estimator 1.14.0
tensorflow-gpu 1.14.0
tensorpack 0.9.8
termcolor 1.1.0
tqdm 4.41.0
urllib3 1.25.7
Werkzeug 0.16.0
wheel 0.33.6
wrapt 1.11.2

Whats wrong with this setup? How can I use this without docker?

EDIT: Looks like I just need the right pb file.. So can you provide the model file for this repo?

from attentionocr.

#57

from attentionocr.