生成三分图的方法及用于生成三分图的神经网络构建方法

摘要

本申请涉及图像处理技术领域，尤其涉及一种生成三分图的方法及用于生成三分图的神经网络构建方法。该生成三分图的方法包括：获取待处理图像；将所述待处理图像输入预设神经网络生成三分图；所述预设神经网络包括主干网络和连接所述主干网络的输出层，所述主干网络包括并联的第一卷积子网络和第二卷积子网络，所述第一卷积子网络和所述第二卷积子网络分别对所述待处理图像进行腐蚀和膨胀，所述输出层用于根据经腐蚀的所述待处理图像和经膨胀的所述待处理图像生成所述三分图。本申请实施例基于神经网络生成待处理图像的三分图，神经网络可以在GPU或NPU上推理，可以大大提升三分图的生成效率。