search

liuheng92 / tensorflow_PSENet

This is a tensorflow re-implementation of PSENet: Shape Robust Text Detection with Progressive Scale Expansion Network.My blog:
https://blog.csdn.net/liuxiaoheng1992/article/details/87646951
MIT License
489 stars 162 forks source link

How to load pre_trained model? #55

Closed ceciliaAI closed 5 years ago

ceciliaAI commented 5 years ago

I used your code to train my own dataset. Now I want to load the trained model for fine-tuning. The currently saved model has models of .index and .data types. How to load it?

liuheng92 commented 5 years ago

use tf.train.Saver, you can read tensorflow api for more infos