CascadePSP

收录理由

做图像分割的人大概都有过这种体验：模型输出的掩膜整体看起来不错，放大一看，边缘却毛毛糙糙。CascadePSP就是为了收拾这种残局而设计的，它把已有二值掩膜和原始图像放在一起，先走一个全局步骤保住整体轮廓，再在局部高分辨率区域里细抠边缘，让掩膜边界立整不少。而且它不挑类别，无论你用的分割器或抠图模型是什么，都能套上去当后处理用。仓库里封装好了pip包，安装后两行代码就能把粗糙的掩膜磨一遍；同时还提供了预训练权重、完整训练脚本，以及一个标注好的4K数据集BIG，想重新训练或者深入研究都有材料可依。要是你不想改动主模型，又想让输出更精细，这算是个成本很低的升级路径。