[发明专利]图像处理模型的训练方法和装置、电子设备及介质

摘要

本申请实施例提供了一种图像处理模型的训练方法和装置、电子设备及介质，属于人工智能技术领域。该方法包括：获取样本图像；对样本图像进行数据增强处理，得到第一图像；将第一图像输入至预设的神经网络模型中，其中，神经网络模型包括轻量化网络、池化网络、解码网络；通过轻量化网络对第一图像进行特征提取，得到样本图像特征图；通过池化网络对样本图像特征图进行池化处理，得到样本池化特征；通过解码网络对样本池化特征进行概率计算，得到样本字段概率值；根据样本字段概率值对神经网络模型的损失函数进行优化，得到图像处理模型。本申请实施例能够有效地减小模型大小，使得模型能够被部署在移动端设备上，提高了模型的适用性。