syyxsxx 5 anos atrás
pai
commit
a9381bef29
1 arquivos alterados com 2 adições e 2 exclusões
  1. 2 2
      deploy/cpp/src/paddlex.cpp

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

@@ -353,7 +353,7 @@ bool Model::predict(const cv::Mat& im, DetResult* result) {
       box->mask.shape = {static_cast<int>(box->coordinate[2]),
                          static_cast<int>(box->coordinate[3])};
       auto begin_mask =
-          output_mask.begin() + (i * classes + box->category_id) * mask_pixels;
+          output_mask.data() + (i * classes + box->category_id) * mask_pixels;
       cv::Mat bin_mask(result->mask_resolution,
                      result->mask_resolution,
                      CV_32FC1,
@@ -362,7 +362,7 @@ bool Model::predict(const cv::Mat& im, DetResult* result) {
                bin_mask,
                cv::Size(box->mask.shape[0], box->mask.shape[1]));
       cv::threshold(bin_mask, bin_mask, 0.5, 1, cv::THRESH_BINARY);
-      auto mask_int_begin = bin_mask.data;
+      auto mask_int_begin = reinterpret_cast<int*>bin_mask.data;
       auto mask_int_end =
         mask_int_begin + box->mask.shape[0] * box->mask.shape[1];
       box->mask.data.assign(mask_int_begin, mask_int_end);