Comments (1)
path = "YOUR/CHECKPOINT/PATH"
OLD_CHECKPOINT_FILE = path + "model_final.ckpt"
NEW_CHECKPOINT_FILE = path + "model2_final.ckpt"
import tensorflow as tf
vars_to_rename = {
"lstm_o/rnn/basic_lstm_cell/weights": "lstm_o/rnn/basic_lstm_cell/kernel",
"lstm_o/rnn/basic_lstm_cell/biases" : "lstm_o/rnn/basic_lstm_cell/bias",
}
new_checkpoint_vars = {}
reader = tf.train.NewCheckpointReader(OLD_CHECKPOINT_FILE)
for old_name in reader.get_variable_to_shape_map():
print old_name
if old_name in vars_to_rename:
new_name = vars_to_rename[old_name]
else:
new_name = old_name
new_checkpoint_vars[new_name] = tf.Variable(reader.get_tensor(old_name))
init = tf.global_variables_initializer()
saver = tf.train.Saver(new_checkpoint_vars)
with tf.Session() as sess:
sess.run(init)
saver.save(sess, NEW_CHECKPOINT_FILE)
from text-detection-ctpn.
Related Issues (20)
- 百度网盘的权重链接失效了,有哪位大佬能提供下训练好的权重文件吗? HOT 4
- training with images from google
- demo.py中 将box转换为原始尺寸时,代码可能有一点小错误
- Crop detected region with result from .txt file but I got the wrong region? HOT 1
- loss值不下降,检测为0
- 请问有data 数据的百度云链接吗? 过期了 HOT 3
- It does not work with tensorflow 2.3.1
- bug in utils.py
- Resource Exhausted: OOM issue
- how to create our own training model? HOT 1
- when i train model My gpu utilization is nearly 0
- ImportError: cannot import name 'bbox' from partially initialized module 'lib.utils' (most likely due to a circular import) (/home/barry/Github/chinese_ocr/ctpn/lib/utils/__init__.py) HOT 3
- 如何预测anchor在X方向上的偏移?有人实现了么,当前模型检测出的文本行,两边会有多余的边缘,在识别的时候,多出的部分严重影响识别效果,
- 运行spilt_label.py结果一直显示Error processing HOT 1
- File "nms.pyx", line 28, in bbox.nms.nms ValueError: Buffer dtype mismatch, expected 'intp_t' but got 'long'
- there are some problem in check[oints HOT 1
- Model Quantization
- 请问怎样把检测出来的长文本进行字符切割?
- Required Python Version 2.7
- during multi-language model training, accuracy of English/numbers is worse than Chinese characters, how to improve?
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from text-detection-ctpn.