forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update import and fix some codes 

remove the  useless parameters  and  update import
This commit is contained in:
shaohua.zhang 2020-07-27 16:11:14 +08:00 committed by GitHub
parent 9af6ab8f89
commit 82a48a7216
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ppocr/utils/save_load.py
View File

@ -21,7 +21,6 @@ import os
import shutil import shutil
import tempfile import tempfile
import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
from .utility import initial_logger from .utility import initial_logger
@ -58,7 +57,7 @@ def _load_state(path):
return state return state
def load_params(exe, prog, path, ignore_params=[]): def load_params(prog, path, ignore_params=None):
""" """
Load model from the given path. Load model from the given path.
Args: Args:
@ -113,14 +112,12 @@ def init_model(config, program, exe):
path = checkpoints path = checkpoints
fluid.load(program, path, exe) fluid.load(program, path, exe)
logger.info("Finish initing model from {}".format(path)) logger.info("Finish initing model from {}".format(path))
return
pretrain_weights = config['Global'].get('pretrain_weights') pretrain_weights = config['Global'].get('pretrain_weights')
if pretrain_weights: if pretrain_weights:
path = pretrain_weights path = pretrain_weights
load_params(exe, program, path) load_params(exe, program, path)
logger.info("Finish initing model from {}".format(path)) logger.info("Finish initing model from {}".format(path))
return
def save_model(program, model_path): def save_model(program, model_path):