InSPyReNet

收录理由

InSPyReNet 能自动找出画面里最抢眼的主体，并把主体与背景干净地分开，而且它专门为高分辨率图像设计，训练时却不需要高分辨率数据，这一点在实际项目中很省事。这个仓库是 ACCV 2022 论文的官方 PyTorch 实现，论文提出了一种图像金字塔结构，配合金字塔混合策略来整合不同尺度的预测结果。代码里直接提供了预训练权重、模型库和简单的推理入口，拿来就能跑。作者还在 DIS5K 二值分割数据集上训练过模型，所以它不只是做显著性检测，做背景移除、图像抠图也顺手。做照片编辑、自动裁剪缩略图，或者搭内容感知的预处理流程，都可以拿它当基线来改。研究者想复现论文里的 benchmark 结果，用仓库自带的权重就能直接跑出来。