一种基于模型轻量化的高精度图像二值分割方法及系统

    公开(公告)号:CN119131394A

    公开(公告)日:2024-12-13

    申请号:CN202411257808.8

    申请日:2024-09-09

    Applicant: 南开大学

    Abstract: 本发明涉及图像处理技术领域,提供了一种基于模型轻量化的高精度图像二值分割方法及系统,包括:获取待分割图像。对于待分割图像,通过学生网络进行分割,得到二值图像;其中,学生网络的训练方法为:将训练集中的原始图像同时作为教师网络和学生网络的输入,采用知识蒸馏策略,通过监督残差连接块得到的特征图,使教师网络指导学生网络进行训练;所述教师网络和学生网络均采用基于双边参考框架的二值分割模型,所述教师网络以Swin‑Large作为骨架网络,所述学生网络以Swin‑Tiny作为骨架网络。使得模型参数量减少的同时图像分割性能损失降低到最小。

Patent Agency Ranking