V2EX 导航·分享创造

精选收藏
按标签筛选:工程杠杆
清除筛选
如果你做过 segmentation,可能默认用了太久 argmax

产品主要功能:rankseg项目用于改进图像分割模型的后处理步骤,通过替换默认的`probs.argmax(dim=1)`为更适合分割目标的策略,以提高整体性能。 描述:rankseg项目旨在优化图像分割流程中的后处理环节,通过改进概率图到最终分割掩码的转换方式,提升分割性能。该项目兼容PyTorch原生概率图流程和Transformers链路,并提供了现成的仓库、文档和可运行的入口,适合进行实验尝试。