浏览代码

solve conflict

will-jl944 4 年之前
父节点
当前提交
76cb346e5f
共有 2 个文件被更改,包括 0 次插入4 次删除
  1. 0 1
      paddlex/cv/models/load_model.py
  2. 0 3
      paddlex/deploy.py

+ 0 - 1
paddlex/cv/models/load_model.py

@@ -51,7 +51,6 @@ def load_model(model_dir, **params):
     Load saved model from a given directory.
     Args:
         model_dir(str): The directory where the model is saved.
-
     Returns:
         The model loaded from the directory.
     """

+ 0 - 3
paddlex/deploy.py

@@ -35,7 +35,6 @@ class Predictor(object):
                  max_trt_batch_size=1,
                  trt_precision_mode='float32'):
         """ 创建Paddle Predictor
-
             Args:
                 model_dir: 模型路径(必须是导出的部署或量化模型)
                 use_gpu: 是否使用gpu,默认True
@@ -183,7 +182,6 @@ class Predictor(object):
 
     def raw_predict(self, inputs):
         """ 接受预处理过后的数据进行预测
-
             Args:
                 inputs(dict): 预处理过后的数据
         """
@@ -204,7 +202,6 @@ class Predictor(object):
 
     def predict(self, img_file, topk=1, transforms=None):
         """ 图片预测
-
             Args:
                 img_file(List[np.ndarray or str], str or np.ndarray):
                     图像路径;或者是解码后的排列格式为(H, W, C)且类型为float32且为BGR格式的数组。