zhang0jhon / AttentionOCR

Scene text recognition
831 stars 258 forks source link

How to run on cpu ?? #13

Open dont32 opened 4 years ago

zhang0jhon commented 4 years ago

Currently the model default runs on GPU for fast speed, if you want to run cpu, you should try to use tensorflow 1.14 instead of tensorflow-gpu 1.14

dont32 commented 4 years ago

I had used only your ocr model in docker but when i run test.py from source having error : ile "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 "/home/dont/.virtualenvs/python3-cpu/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 3972, in get_tensor_by_name return self.as_graph_element(name, allow_tensor=True, allow_operation=False) File "/home/dont/.virtualenvs/python3-cpu/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 3796, in as_graph_element return self._as_graph_element_locked(obj, allow_tensor, allow_operation) File "/home/dont/.virtualenvs/python3-cpu/lib/python3.5/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." @zhang0jhon anny different ??