1.一种基于改进YOLOX算法的目标检测方法,其特征在于,
获取商标数据集,并将商标数据集内的图片格式转化为可以直接输入YOLOX网络进行训练的格式,得到可用商标数据集;
将可用商标数据集的数据进行预处理,将所有的商标图片划分为训练集、验证集和测试集;
分析训练集、验证集和测试集,确定所有的商标种类;
基于YOLOX的网络结构进行改进,得到改进后的YOLOX目标检测算法;
用训练数据集对改进后的YOLOX目标检测算法进行训练,评估得到最佳训练权重文件,得到最终YOLOX目标检测算法;
所述基于YOLOX的网络结构进行改进,得到改进后的YOLOX目标检测算法,包括:对YOLOX的网络结构的特征提取骨干网络进行改进,将YOLOX的网络结构中的空间金字塔模块SPP替换为ASPP,在YOLOX网络结构的特征金字塔模块的小目标分支后添加改进型感受野模块RFB-s,将改进型感受野模块RFB-s放入80*80的分类器之前;
所述小目标分支即80*80特征图分支;
其中,所述改进型感受野模块RFB-s中采用3×3卷积层代替5×5卷积层,并采用1×3和3×1卷积层代替3×3卷积层。
2.根据权利要求1所述的基于改进YOLOX算法的目标检测方法,其特征在于,所述获取商标数据集,并将商标数据集内的图片格式转化为可以直接输入YOLOX网络进行训练的格式,包括:将商标数据集内的图片转化为PNG格式图片,将商标数据集内的记载的图片数据统一转化为VOC数据集格式。
3.根据权利要求1所述的基于改进YOLOX算法的目标检测方法,其特征在于,所述将可用商标数据集的数据进行预处理,将所有的商标图片划分为训练集、验证集和测试集,包括:将可用商标数据集的数据按6:2:2的比例划分训练集、验证集和测试集。
4.根据权利要求2所述的基于改进YOLOX算法的目标检测方法,其特征在于,所述分析训练集、验证集和测试集,确定所有的商标种类,包括:统计所有商标的种类个数,统计商标的名称,将商标的名称写入到VOC数据集内。
5.根据权利要求1所述的基于改进YOLOX算法的目标检测方法,其特征在于,所述用训练数据集对改进后的YOLOX目标检测算法进行训练,评估得到最佳训练权重文件,得到最终YOLOX目标检测算法,包括:用训练数据集对改进后的YOLOX目标检测算法进行训练,进行300个批次的训练之后取其中最佳的训练权重文件,最佳的训练权重文件加载到改进后的YOLOX目标检测算法中,得到最终YOLOX目标检测算法。