利用百度飞桨PaddlePaddle框架搭建一个卷积神经网络,对包含斑马线的马路和不包含斑马线的马路图像进行分类。数据集中一张图像的样本如图3.25所示。
图3.25 图像样本示例
卷积神经网络具体设计流程如图3.26所示,大致分为数据处理、模型设计、训练配置、训练过程、模型保存这几个步骤,每个步骤中又包含各自模型不同的小细节。
由于项目代码篇幅较长,本节不附项目代码及介绍,读者可以扫码下载或通过链接在百度AI Stuido线上运行,项目代码附有注释,且配套有视频讲解。
项目运行:斑马线检测
https://aistudio.baidu.com/projectdetail/5107670?contributionType=1
图3.26 卷积神经网络设计流程