diff --git a/inference/README.md b/inference/README.md index 5e65246dc1053b7f671a91ff11e7b00021edeb31..84138aa0c075f159c8f6feb305694bb011c2b1a2 100644 --- a/inference/README.md +++ b/inference/README.md @@ -22,7 +22,7 @@ inference ├── images │ └── humanseg # 示例人像分割模型测试图片目录 ├── tools -│ └── visualize.py # 示例人像分割模型结果可视化脚本 +│ └── visualize.py # 示例分割模型结果可视化脚本 ├── docs | ├── linux_build.md # Linux 编译指南 | ├── windows_vs2015_build.md # windows VS2015编译指南 @@ -128,9 +128,7 @@ D:\projects\PaddleSeg\inference\build\Release>demo.exe --conf=D:\\projects\\Padd | input_dir | 需要预测的图片目录 | -配置文件说明请参考上一步，样例程序会扫描input_dir目录下的所有图片，并生成对应的预测结果图片： - -文件`demo.jpg`预测的结果存储在`demo_jpg.png`中，可视化结果在`demo_jpg_scoremap.png`中， 原始尺寸的预测结果在`demo_jpg_recover.png`中。 +配置文件说明请参考上一步，样例程序会扫描input_dir目录下的所有图片，并生成对应的预测结果。图像分割会对`demo.jpg`的每个像素进行分类，其预测的结果保存在`demo_jpg.png`中。分类预测结果的图不能直接看到效果，必须经过可视化处理。对于二分类的图像分割模型，样例程序自动将预测结果转换成可视化结果，保存在`demo_jpg_scoremap.png`中， 原始尺寸的预测结果在`demo_jpg_recover.png`中，如下图。对于**多分类**的图像分割模型，请参考[可视化脚本使用方法](./docs/vis.md)。 输入原图 ![avatar](images/humanseg/demo2.jpeg) diff --git a/inference/docs/demo.jpg b/inference/docs/demo.jpg new file mode 100644 index 0000000000000000000000000000000000000000..8e5f2157e2e5ae9628e764ef3585461fc50841fd Binary files /dev/null and b/inference/docs/demo.jpg differ diff --git a/inference/docs/demo_jpg.png b/inference/docs/demo_jpg.png new file mode 100644 index 0000000000000000000000000000000000000000..cc702a6900eb48903794adc601821b3717d2ff68 Binary files /dev/null and b/inference/docs/demo_jpg.png differ diff --git a/inference/docs/vis.md b/inference/docs/vis.md new file mode 100644 index 0000000000000000000000000000000000000000..c694518765d907fde6c55aaae864b5ce29ccb5c5 --- /dev/null +++ b/inference/docs/vis.md @@ -0,0 +1,30 @@ +# 图像分割结果可视化说明 + +本文档介绍如何使用可视化脚本对图像分割结果进行可视化处理。**注意：可视化脚本处理仅针对PaddleSeg C++预测部署方案生成的预测结果。** + +## 说明 +图像分割模型会对预测的图像的每个像素进行分类，所以图像分割模型的预测结果是图像里每个像素的标签，我们将预测结果以图片格式保存。例如预测图片`demo.jpg`，其预测的结果以图片格式保存在`demo_jpg.png`中。保存分割预测结果的图片并不能直接看到效果（一张全黑的图），必须经过可视化处理。以下为使用方法。 + +```bash +# 当前路径为PaddleSeg根目录 +cd inference/tools/ +# 拷贝保存分割预测结果的图片到本目录 +cp XXX/demo_jpg.png . +# 运行可视化脚本 +python visualize.py demo_jpg.png vis_result.png +``` +该脚本中第一个参数为保存预测结果的图片的路径，第二个参数为可视化后的效果图路径。 + +**可视化结果展示：** + +以下以cityscape模型的预测结果进行展示。 + +原图(`demo.jpg`): +![avatar](demo.jpg) + +预测结果图(`demo_jpg.png`): +![avatar](demo_jpg.png) + +效果图(`vis_result.png`)： +![avatar](vis_result.png) + diff --git a/inference/docs/vis_result.png b/inference/docs/vis_result.png new file mode 100644 index 0000000000000000000000000000000000000000..0b342a3d4d90bdf088c06880164b9ccf9b2cbec1 Binary files /dev/null and b/inference/docs/vis_result.png differ