一种基于Transformer和CNN的双分支网络的图像分割方法及系统

    公开(公告)号:CN117710381A

    公开(公告)日:2024-03-15

    申请号:CN202311697900.1

    申请日:2023-12-11

    Applicant: 无锡学院

    Abstract: 本发明公开了一种基于Transformer和CNN的双分支网络的图像分割方法及系统,方法包括以下步骤:获取芯片X光图像,并对芯片X光图像进行预处理;建立交互式双分支网络模型,所述交互式双分支网络模型由CNN分支网络、PVTv2模型、融合模块,CNN分支网络、Transformer分支网络的输出端分别连接有语义分割头和辅助分割头;其中,CNN分支网络由通道压缩的Resnet18模型为基础框架,Resnet18模型包括第一残差模块‑第四残差模块,第一残差模块包括Stem模块和两个残差块Res block,第二残差模块‑第四残差模块均包括最大池化层和两个残差块Res block。本发明建立基于Transformer分支网络和CNN分支网络的交互式双分支网络模型,从输入端开始建立图像全局和局部之间的信息交流,提升芯片X光图像的分割精度。

Patent Agency Ranking