一种基于多任务学习的沿海养殖池语义分割方法

本发明公开一种基于多任务学习的沿海养殖池语义分割方法，属于遥感影像处理技术领域，用于沿海养殖池智能提取，包括在统一的框架下对沿海养殖池的掩膜、边缘以及距离同时进行学习，通过任务之间的权重共享，有效提升沿海养殖池提取结果的准确性；以改进的ConvNeXt V2块作为网络构建单元，组成单编码器‑解码器的多任务网络，在保证网络轻量化的同时，充分发挥了多任务网络的性能；以Transformer多头注意力机制为参考，设计面向多任务卷积网络的层次化多头注意力，针对不同的任务的特性，对特征进行有效的增强。对比现有技术，本发明提高了单个任务的性能表现，解决了边界粘连问题，提升了提取精确度和模型泛化性。