update paddle.load

Author: CycloneBoy

ppdettorch/process/transform/convert_paddle_to_torch_base.py

@@ -9,6 +9,8 @@
 import os
 import paddle
 
+from ppdettorch.utils.model_utils import ModelUtils
+
 
 class ConvertPaddleToTorchBase(object):
 
@@ -24,7 +26,8 @@ def load_paddle_model(file_name, show_info=True):
         :param show_info:
         :return:
         """
-        paddle_model_params = paddle.load(file_name, return_numpy=True)
+        # paddle_model_params = paddle.load(file_name, return_numpy=True)
+        paddle_model_params = ModelUtils.load(file_name, return_numpy=True)
 
         model_params = []
         for weight_name, weight_value in paddle_model_params.items():

ppdettorch/utils/constant.py

@@ -13,7 +13,7 @@ class Constants(object):
 
     """
     USER_HOME = os.path.expanduser("~")
-    WORK_DIR = f"{USER_HOME}/shenglei/ocr/PPDetectionPytorch"
+    WORK_DIR = f"{USER_HOME}/work/PPDetectionPytorch"
     # WORK_DIR = "./"
 
     DATA_DIR = f"{WORK_DIR}/data"