在水利场景的语义分割任务中，目标复杂多样且需要保证一定的实时性。针对这些特点，本文提出了一种轻量级的水利场景语义分割算法，该方法在BiSeNet(bilateral segmentation network)网络基础上，引入嵌入SE注意力机制的MobileNet v2作为主干，降低模型的大小，提高检测的实时性；并在主干末端增加一个金字塔池化层，得到增强的深层特征，提高模型的分割准确度。实验表明该方法在水利数据集的MIoU表现，比改进前高出4.66%，模型参数也大大减少。
科研之友
微信
新浪微博
Facebook
分享链接
