树莓派上利用Tensorflow实现小车的自动驾驶

作者: Timthony 2018-11-30 17:53:51

先抛出大家最关心的——代码地址:

github传送门:https://github.com/Timthony/self_drive

码云传送门:https://gitee.com/tiantianhang/self_drive

基于树莓派的人工智能自动驾驶小车

# 整体流程

  1. 电机控制
  2. 摄像头调试
  3. 道路数据采集
  4. 搭建深度学习模型,参数调试
  5. 自动驾驶真实道路模拟
  6. 参数最终调试

使用方法:

  1. 先将树莓派小车硬件组装好
  2. 使用zth_car_control.py来控制小车的前后左右移动,配合zth_collect_data.py来人工操作,使小车在自己制作的跑道进行数据采集。(该过程在树莓派进行)
  3. 数据采集完成以后使用zth_process_img.py来对采集的数据进行处理,之前当前先完成一些数据清洗的工作。(电脑上执行)
  4. 使用神经网络模型对数据进行训练zth_train.py,得到训练好的模型。(电脑上执行)
  5. 在树莓派小车上使用zth_drive和训练好的模型,载入模型,即可实现在原先跑道的自动驾驶。(树莓派上执行)

注意:只需要使用上述提到的代码即可,别的都是一些初始版本或者正在增加的一些新模块。

树莓派上利用Tensorflow实现小车的自动驾驶

树莓派上利用Tensorflow实现小车的自动驾驶

# 注意事项:

  1. 赛道需要自己制作,很重要,决定了数据质量。(我是在地板上,贴的有色胶带,然后贴成了跑道的形状)。
  2. 赛道的宽度大约是车身的两倍。
  3. 大约采集了五六万张图像,然后筛选出三四万张。
  4. 摄像头角度问题

# 具体制作流程:

  1. 小车原始模型,某宝购买玩具车即可,比如:有电机,有自带电池盒(给电机供电)
  2. 树莓派,摄像头,蓄电电池组(用于树莓派供电)
  3. 使用一些螺栓,螺柱,亚克力板将树莓派,蓄电电池固定在小车上(具体方法,看手头的工具吧)
  4. 组装好以后,树莓派通过VNC连接电脑,登陆树莓派,在树莓派安装keras环境,以便***调用训练好的模型。
  5. 关于小车的控制(电机控制,摄像头采集数据),都在源文件,有注释,大致思路就是通过方向键AWSD来控制方向,使用了pygame的工具包。
  6. 通过电脑端的wasd方向键手动控制小车(已经VNC连接好)在制作好的赛道上进行图像采集,直线部分按w,左拐弯按a,右拐弯按d等,建议采集50000张以上。

(采集的图像命名要求为,0_xxxx,1_xxxx,其中首位字母就代表了你按下的是哪个键,比如图像是0开头,那么这张图像就是直行,按下的是w键,这些0,1,2,3,4 数字就相当于数据的标签值)

       7. 将图片从树莓派拷贝下来,进行数据清洗,使用电脑端的深度学习环境进行模型训练,使用的模型可以自行定义。

       8. 将训练好的模型文件.h5拷贝到树莓派,然后通过树莓派调用载入模型,即可处理实时的图像,并且根据图像预测出是0,1,2,3,4等数字,也就表示了树莓派该怎么移动,通过树莓派控制电机即可。

# 正在进行一些改进:

  1. 使用迁移学习进行fine-tuning是否可以提高精度
  2. 处理光照问题
  3. 处理数据类别不平衡的问题

欢迎交流讨论

树莓派 Tensorflow 自动驾驶
上一篇:新一代大数据与人工智能基础架构技术的发展与趋势 下一篇:数据库运维的那些难题,我们用机器学习解决了
评论
取消
暂无评论,快去成为第一个评论的人吧

更多资讯推荐

人工智能遇冷,自动驾驶受阻?

2016到2019,人工智能经历了梦幻般的三年,但人工智能的历史规律告诉我们:高潮过后可能会引来新的一波沉寂,人工智能助推下的自动驾驶也会受到波及。

佚名 ·  3天前
2019机器学习框架之争:与Tensorflow竞争白热化,进击的PyTorch赢在哪里?

2019年,机器学习框架之争进入了新阶段:PyTorch与TensorFlow成为最后两大玩家,PyTorch占据学术界领军地位,TensorFlow在工业界力量依然强大,两个框架都在向对方借鉴,但是都不太理想。

大数据文摘 ·  2019-10-11 23:18:15
5G、自动驾驶、人工智能都到什么阶段了?一张图就能看懂

今年,Gartner从2000项技术中选出了29项,并由此总结出了企业决策者应该纳入考虑范围的五大创新技术趋势。Gartner研究副总裁Brian Burke在接受CIO Dive采访时指出,「人工智能渗透到了其他所有趋势中。」

biu ·  2019-09-25 08:20:58
自动驾驶列车已准备就绪!但你敢坐吗?

如果计划进展顺利,矿业巨头必和必拓(Rio Tinto)的一列货运列车将于8月份开始在美国科罗拉多州普韦布洛(Pueblo)附近的轨道上轰隆隆地驶过。最特别的是,这列火车完全由电脑控制。

Aaron Pressman ·  2019-08-01 13:40:38
北京市首个T5级别自动驾驶封闭测试场正式开放运营

近日,7月8日消息,继国家智能汽车与智慧交通(京冀)示范区海淀基地后,国家智能汽车与智慧交通(京冀)示范区亦庄基地(以下简称亦庄基地)近期通过相关部门审批,封闭测试场正式对外开放运营。

佚名 ·  2019-07-09 12:45:21
车联网频率三年免费 自动驾驶产业化提速

近日,国家发改委、财政部发布通知称,为鼓励新技术新业务的发展,对5905-5925MHz频段车联网直连通信系统频率占用费标准实行“头三年免收”优惠政策,即自频率使用许可证发放之日起,第一至第三年免收无线电频率占用费,第四年及以后按照国家规定的收费标准收取频率占用费。

易欢 ·  2019-06-25 15:47:58
CVPR2019, 百度Apollo首次曝光L4级自动驾驶纯视觉解决方案Apollo Lite

美国当地时间6月16日-20日,全球计算机视觉及模式识别领域顶级学术会议CVPR 2019(Conference on Computer Vision and Pattern Recognition)于美国长滩召开。

佚名 ·  2019-06-19 15:04:43
自动驾驶真的来了!全国七地打响第一枪

近些年来,唯一毫无争议被视为汽车行业具有变革意义的新技术,估计只有自动驾驶了。无人驾驶技术如果可以得到全面普及,将会大大节约人类生产生活的时间成本,同时在某种程度上也会让驾驶行为变得更加安全。

汽车华尔街 ·  2019-06-12 07:50:23
Copyright©2005-2019 51CTO.COM 版权所有 未经许可 请勿转载