Explorar o código

inputs_batch_.size() -> batch_size

jack %!s(int64=5) %!d(string=hai) anos
pai
achega
214a8e5c82
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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;