liuheng92 / tensorflow_PSENet

This is a tensorflow re-implementation of PSENet: Shape Robust Text Detection with Progressive Scale Expansion Network.My blog:

Home Page:https://blog.csdn.net/liuxiaoheng1992/article/details/87646951

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to load pre_trained model?

ceciliaAI opened this issue · comments

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?

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