基于改进BiSeNet的轻量级水利语义分割算法

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