一种嵌入式端的表计主刻度线分割方法

    公开(公告)号:CN112614135A

    公开(公告)日:2021-04-06

    申请号:CN202011537925.1

    申请日:2020-12-23

    Abstract: 本发明涉及图像处理和神经网络领域,提供一种嵌入式端的表计主刻度线分割方法,所述嵌入式端的表计主刻度线分割方法包括:步骤1,构建用于主刻度线分割任务的原始Unet网络作为知识蒸馏的教师网络;步骤2,将原始Unet网络进行模型压缩,得到Thin‑Unet网络;步骤3,对Thin‑Unet网络进行卷积替换方式获得一个参数量较小的Unet网络,并训练得到作为知识蒸馏的学生网络;步骤4,采用知识蒸馏方法,将教师网络与学生网络连接起来进行训练,不断更新学生网络的参数,得到参数量接近教师网络参数量的学生网络;步骤5,利用训练完成的学生网络进行主刻度线分割。本发明的技术方案能够在保持精度接近的情况下,大大降低了模型的计算量,使其能部署在嵌入式端运行。

Patent Agency Ranking