From ff90e5a04f8b5a8c4d0fd8a4b79171f349f2cac1 Mon Sep 17 00:00:00 2001
From: syyxsxx <csuwsy@163.com>
Date: Mon, 31 Aug 2020 16:00:12 +0800
Subject: [PATCH] fix mask

---
 deploy/cpp/src/visualize.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/deploy/cpp/src/visualize.cpp b/deploy/cpp/src/visualize.cpp
index a0caf9c..3443848 100644
--- a/deploy/cpp/src/visualize.cpp
+++ b/deploy/cpp/src/visualize.cpp
@@ -84,9 +84,11 @@ cv::Mat Visualize(const cv::Mat& img,
     if (boxes[i].mask.data.size() == 0) {
       continue;
     }
+    std::vector<float> mask_data;
+    mask_data.assign(boxes[i].mask.data.begin(), boxes[i].mask.data.end());
     cv::Mat bin_mask(boxes[i].mask.shape[1],
                      boxes[i].mask.shape[0],
-                     CV_8UC1,
+                     CV_32FC1,
                      boxes[i].mask.data.data());
     cv::Mat full_mask = cv::Mat::zeros(vis_img.size(), CV_8UC1);
     bin_mask.copyTo(full_mask(roi));
-- 
GitLab