collision_avoidance
時(shí)間:2022-03-16 16:12:02 | 來(lái)源:行業(yè)動(dòng)態(tài)
時(shí)間:2022-03-16 16:12:02 來(lái)源:行業(yè)動(dòng)態(tài)
目錄下使用就行。文件鏈接如下:https://drive.google.com/file/d/1UsRax8bR3R-e-0-80KfH2zAt-IyRPtnW/view
由于我們未得到原創(chuàng)團(tuán)隊(duì)的授權(quán),不能擅自下載這個(gè)存放在Google網(wǎng)盤(pán)上的文件,再分享給讀者下載,請(qǐng)大家能夠理解,這需要讀者請(qǐng)自行設(shè)法下載!
接下來(lái)開(kāi)啟notebooks/collision_avoidance/live_demo.ipynb工作腳本,逐步執(zhí)行就能讓Jetbot小車(chē)執(zhí)行避障的功能。這里面主要分為以下三大部分:
- 加載訓(xùn)練的模型:
這里關(guān)于深度學(xué)習(xí)的部分,全部使用PyTorch這個(gè)輕量級(jí)的框架,對(duì)于不熟悉的讀者來(lái)說(shuō),一開(kāi)始的兩行代碼可能就已經(jīng)會(huì)產(chǎn)生不小的困擾,現(xiàn)在就簡(jiǎn)單地逐行說(shuō)明:
| model = torchvision.models.alexnet(pretrained=False)
|
- torchvision是PyTorch里面專(zhuān)門(mén)用在視覺(jué)應(yīng)用的深度學(xué)習(xí)庫(kù)。
- 由于PyTorch內(nèi)建支持很多常用的神經(jīng)網(wǎng)絡(luò)結(jié)構(gòu),這里要使用深度卷積網(wǎng)絡(luò)始祖的AlexNet,是因?yàn)榍懊嫣峁┫螺d連接的