用于目标检测的神经网络训练方法、装置、设备、介质
摘要:
本发明提供用于目标检测的神经网络训练方法、装置、设备、介质,该方法包括:S1:将检测目标图像输入teacher网络、student网络;S2:将所述teacher网络中第四层的第一特征值、所述student网络中的第四层的第二特征值,进行比较,得到比对结果;S3:利用所述比对结果,对所述student网络进行调整,并进入步骤S2,直到得到的比对结果小于预设阈值,以得到训练后的student网络;其中,所述teacher网络与所述student网络的总体构架相同;所述teacher网络处于前向预测模式,所述student网络处于训练模式;所述teacher网络的通道数比所述student网络的通道数多;所述检测目标图像,设有检测目标的矩形框以及所述检测目标的标签类别。本发明针对检测目标图像,占用计算资源小。
0/0