应用于图像处理的神经网络训练方法及相关设备

摘要文本

荣耀终端有限公司取得“一种透气窗帘布”专利技术，本申请实施例涉及人工智能领域，提供一种应用于图像处理的神经网络训练方法及相关设备，该方法包括：构建与处理任务对应的超网络，超网络可包括多个子网络，多个子网络与超网络共享网络参数；对超网络进行采样训练；基于训练后的超网络的网络参数，确定目标子网络，目标子网络可为超网络中的子网络或者由超网络中的子网络变异得到，且目标子网络的性能符合预设要求；对目标子网络进行量化训练，得到量化后的目标子网络，在量化训练过程中，目标子网络与超网络共享量化训练使用的量化器。利用本申请的目标子网络与超网络共享量化器，能够在量化训练中实现网络结构搜索中的蒸馏作用，可提升最终部署在终端设备侧的目标子网络的图像处理效果。