使用深度学习在Unity环境中训练Donkey Car —— 驾驶Donkey Car

前言:

        通过前面两篇博客的介绍,大家现在已经能部署好Unity环境制造数据并且训练自己的模型。

        两篇博客的地址:

        使用深度学习在Unity环境中训练Donkey Car —— 环境搭建和数据制造 - 简书

        使用深度学习在Unity环境中训练Donkey Car —— 训练模型 - 简书

        今天这篇博客则是最后一步,验证自己训练的模型在Unity中驾驶Donkey Car的效果怎么样!(文中有最终效果演示视频链接)

开启服务:

        在使用模型之前需要知道Unity是如何与python通信的!

        由于我们是在python中编写的代码,因此我们必须首先找到一种方法来使Python与Unity环境进行通信。通信是通过Websocket协议完成的。与HTTP不同,Websocket协议允许服务器和客户端之间的双向双向通信。在这里,我们的python“服务器”可以直接将消息推送到Unity(例如转向和节速度),我们的Unity“客户端”也可以将信息(拍摄图片)推送回python服务器。在python“服务器”与Unity传递信息时会涉及解码和编码的工作,如将图片编码成数据流的形式传入“服务器”,“服务器”收到收到数据流会进行一个解码,再将解码得到的信息传入网络,网络输出的数值也会再次转换为数据流传给“服务器”,最终服务器再传给Unity。(具体代码在tcp_server.py文件中)

我们之前训练好的模型在outputs文件夹下,现在就要使用训练好的模型开启服务:

1、首先在“sdsandbox-master”文件夹内打开Power Shall(如图)

打开power shall

2、输入“python src/predict_server.py --model=outputs/mymodel.h5”指令开启服务  (mymodel.h5是我当时保存的模型,每个人自己保存的模型名字可能不一样)

运行python“服务器”

驾驶Donkey Car:

        打开服务器之后再进入Unity环境中,点击开始按钮,再单击右上方的“Use NN Steering”按钮(第三个空白按钮,至于为什么是空白的,我已经在第一篇博客做了解释)。

运行环境,连接服务器

此时Unity会连接上python“服务器”(如图)

建立连接

        这样Donkey Car就通过我们自己训练的模型在道路上行驶。

        我在bilibili上传了一段视频(这段视频演示了 “有模型驾驶Donkey Car” 和 “无模型驾驶Donkey Car” 之间的区别,通过对比可以明显看出有多有趣!!!)。视频链接:“有模型驾驶Donkey Car” 和 “无模型驾驶Donkey Car” 之间的区别

总结:

        首先需要说明的是训练Donkey Car的数据不要过大,过大容易使模型过拟合,会出现开着开着就跑出去了,制造数据的操作重复5次即可。建议大家搭建好GPU环境,因为GPU训练速度快太多了。大家如果想提升速度可以通过对训练数据名修改,将里面的速度值提高到2或3,同时在启动服务器之前需要将predict_server.py 中的默认速度调成与训练速度值一致(如训练数据为2,默认速度也改为2).

        在Unity中训练Donkey Car到此就结束了,整个过程不难,十分有趣,大家可以在业余时间娱乐娱乐,有闲钱的小伙伴也可以买个真的Donkey Car在现实环境中玩玩!!

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,080评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,422评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,630评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,554评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,662评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,856评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,014评论 3 408
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,752评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,212评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,541评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,687评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,347评论 4 331
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,973评论 3 315
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,777评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,006评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,406评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,576评论 2 349

推荐阅读更多精彩内容