浏览代码

inputs_batch_.size() -> batch_size

jack 5 年之前
父节点
当前提交
214a8e5c82
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      deploy/cpp/src/paddlex.cpp

+ 1 - 1
deploy/cpp/src/paddlex.cpp

@@ -111,7 +111,7 @@ bool Model::preprocess(const cv::Mat& input_im, ImageBlob* blob) {
 
 // use openmp
 bool Model::preprocess(const std::vector<cv::Mat> &input_im_batch, std::vector<ImageBlob> &blob_batch) {
-  int batch_size = inputs_batch_.size();
+  int batch_size = input_im_batch.size();
   bool success = true;
   int max_h = -1;
   int max_w = -1;