一种用于多类别垃圾场景识别的目标检测模型的训练方法

摘要

本发明提供了一种用于多类别垃圾场景识别的目标检测模型的训练方法，包括：构造基础目标检测模型；在p个卷积层中，选取连续的m个卷积层，将选中的每个卷积层均替换为空洞卷积；依次设置m个空洞卷积的孔洞率，满足任意相邻的两个空洞卷积的孔洞率的最大公约数为1，以获取优化目标检测模型；使用样本图像训练集对优化目标检测模型训练，以获取用于多类别垃圾场景识别的目标检测模型。本发明解决了现有技术中的目标检测模型因其感受野范围小，从而对具有广阔视野的场景图像存在垃圾目标检测精度低的问题，将此类场景图像作为模型输入进行垃圾识别检测，所得到的检测结果往往会出现多点位垃圾识别无效或垃圾检测误报的现象。