Python-基于卷积神经网络斑马线识别分类，准确率高达99%

最新推荐文章于 2024-05-11 16:25:25 发布

醉公子～

最新推荐文章于 2024-05-11 16:25:25 发布

阅读量7.5k

点赞数 37

分类专栏：深度学习文章标签：深度学习神经网络 tensorflow ncnn 卷积神经网络

本文链接：https://blog.csdn.net/qq_44936246/article/details/116460870

版权

本文介绍了使用Python和Tensorflow构建的卷积神经网络模型进行斑马线识别，准确率高达99%。文章涵盖了神经网络结构、数据集准备、模型搭建、训练过程和过拟合解决方案，旨在帮助读者理解智慧斑马线技术的应用和模型训练技巧。

摘要由CSDN通过智能技术生成

在这里插入图片描述

引言

目前智慧城市的发展，人们生活处处有科技，比如人脸识别，智慧交通，无人驾驶等前沿的科技产品也都融入了人们生活中；
智慧交通的发展也越来越成熟，车辆识别，车辆跟踪，车辆分析，行人跟踪检测与行为分析，这些都越来越成熟，但是都是建立在斑马线的基础上来实现的。

智慧斑马线，严格地说，是用于（单条或区域多条）人行横道（斑马线）上，人、车、路、环境协同的一整套软硬件产品与服务，主要通过提高斑马线标线的清晰度和醒目度，以及智能网联控制，实现智能的斑马线通行的警示、调度、控制、多点联动。

智慧斑马线首先是斑马线。斑马线是一条线，对行人来说是保护道路通行安全的生命线；对城市交通管理来说，是对人、车、路进行协同工作的一条协同线，只有人和车、路都是按规则有秩序地协同，才能达到交通文明有序。智慧斑马线就是以技术为出发点，帮助行人和城市交管做到斑马线生命线和协同线的一种智能产品和服务。

神经网络的结构

神经网络模型包含三个部分：input layer（输入层）、hidden layer（中间层或隐藏层）、output layer（输出层）。其中，hidden layer的层数不固定，在某些简单问题中，hidden layer的层数可能为0，仅有input layer和output layer；在复杂问题中，hidden layer的层数也可能成百上千。
在这里插入图片描述

模型中每一层的节点称为“神经元”。位于input layer的神经元对应着训练数据的特征。hidden layer和output layer中的神经元由activation function（激活函数）表达，我们用字母 g g g表示。Activation function有很多种类型，最常用是sigmoid函数，它的表达式如下：

在这里插入图片描述

sigmoid函数图像如下图所示。当 x > > 0 x>>0 x>>0时， f ( x ) f(x) f(x)无限逼近于1；当 x < < 0 x<<0 x<<0时， f ( x ) f(x) f(x)无限逼近于0；当 x = 0 x=0 x=0时， f ( x ) = 0.5 f(x)=0.5 f(x)=0.5。

搭建模型

一、引入基本库

环境配置：

Windows10系统
Python–3.7.3
tensorflow–2.3
OpenCV–4.5.1

import tensorflow as tf
import matplotlib.pyplot as plt
import cv2 as cv
import pandas as pd
import numpy as np
import os
import shutil
from keras.models import Sequential, load_model
from keras import optimizers
from keras.preprocessing.image