提交 ff90e5a0 编写于 作者: S syyxsxx

fix mask

上级 860ee161
...@@ -84,9 +84,11 @@ cv::Mat Visualize(const cv::Mat& img, ...@@ -84,9 +84,11 @@ cv::Mat Visualize(const cv::Mat& img,
if (boxes[i].mask.data.size() == 0) { if (boxes[i].mask.data.size() == 0) {
continue; 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], cv::Mat bin_mask(boxes[i].mask.shape[1],
boxes[i].mask.shape[0], boxes[i].mask.shape[0],
CV_8UC1, CV_32FC1,
boxes[i].mask.data.data()); boxes[i].mask.data.data());
cv::Mat full_mask = cv::Mat::zeros(vis_img.size(), CV_8UC1); cv::Mat full_mask = cv::Mat::zeros(vis_img.size(), CV_8UC1);
bin_mask.copyTo(full_mask(roi)); bin_mask.copyTo(full_mask(roi));
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册