未验证 提交 8750b689 编写于 作者: B Bin Long 提交者: GitHub

Fix inference mask saving bug

上级 7b83d4dd
......@@ -97,7 +97,7 @@ namespace PaddleSolution {
cv::Mat mask_png = cv::Mat(eval_height, eval_width, CV_8UC1);
mask_png.data = _mask.data();
std::string nname(fname);
auto pos = fname.find(".");
auto pos = fname.rfind(".");
nname[pos] = '_';
std::string mask_save_name = nname + ".png";
cv::imwrite(mask_save_name, mask_png);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册