在Alexnet网络模型的基础上,基于迁移学习原理,来训练R-CNN目标检测网络。并对Matlab自带的stop sign(停止标志)图像数据集进行识别,该数据集已标注好。其实现步骤如下:
步骤一:导入Alexnet预训练的模型
步骤二:载入训练集图像
步骤三:设置训练参数,并基于迁移学习原理,在Alexnet卷积神经网络基础上,通过41幅包括stop sign的图像训练R-CNN检测器。
步骤四:使用测试图像检验训练好的目标检测器对stop sign图像的检测效果,在原图上标记目标区域并显示类别和置信度。
1