小编Chi*_*tan的帖子

我想用OpenCV-DNN包装注意力 OCR模型以增加推理时间。我正在使用官方 TF models repo中的 TF 代码。

对于用 OpenCV-DNN 包装 TF 模型，我参考了这段代码。需要cv2.dnn.readNetFromTensorflow()“冻结图”和“图结构”来读取 TF 模型。

我使用此代码片段从元检查点文件导入结构并将图形结构保存在文件中.pbtxt。

# load graph from meta file
tf.reset_default_graph()  
imported_meta = tf.train.import_meta_graph("attention_ocr_2017_08_09/model_demo_inference.ckpt.meta")

# restore graph structure, variables in session's graph
sess = tf.Session()
imported_meta.restore(sess, 'attention_ocr_2017_08_09/model_demo_inference.ckpt')
# write graph structure to a pbtxt file
tf.train.write_graph(sess.graph_def, './', 'train_attention.pbtxt', as_text=True)

冻结图形，代码如下：

from tensorflow.python.tools import freeze_graph
freeze_graph.freeze_graph('train_attention.pbtxt', '', False, \
                          'attention_ocr_2017_08_09/model_demo_inference.ckpt', \
                          'AttentionOcr_v1_1/Softmax', \
                          'save/restore_all', 'save/Const:0', 'frozen_model.pb', True, "")

最终代码使用函数中的pbtxt和文件。 …

我正在寻找一种方法来手动禁用深色主题设置，因为它在查看数据库表时会伤害我的眼睛。我在首选项中找不到任何设置。任何帮助将不胜感激。