MMdetion3D是是mmlab在3d目標檢測方面提供的相關(guān)檢測模型,可以實現(xiàn)點云、圖像或者多模態(tài)數(shù)據(jù)上的3D目標檢測以及點云語義分割。
GitHub地址:https://github.com/open-mmlab/mmdetection3d/
目前mmdetection3d 支持21種不同的算法,100多個預(yù)訓(xùn)練模型,7個數(shù)據(jù)集:
mmdetection3D安裝比較簡單,之前弄的時候踩了很多坑,老是卡在安裝mmcv-full這一步,特別是在win系統(tǒng)下感覺老是有問題。在ubuntu系統(tǒng)下可以直接通過mim來安裝很方便(windows 的 shell其實也可以,不過沒搞透)。具體步驟如下(ubuntu系統(tǒng)或者win的shell):
創(chuàng)建conda環(huán)境
conda create -n mmdet3d python=3.7 pytorch==1.6.0 cudatoolkit=10.1 torchvision -c pytorch -y
這邊要注意一點就是需要根據(jù)自身計算機的cuda版本來選擇,這個很重要?。。?br> 然后就是激活環(huán)境進行安裝了,推薦使用mim進行安裝,比較方便
conda activate mmdet3d
pip install openmim
mim install mmcv-full
git clone https://github.com/open-mmlab/mmdetection3d.git
cd mmdetection3d-master
pip install -e .
需要注意的一點是有時候會在安裝mmcv-full的時候出現(xiàn)問題。這個時候就需要到官網(wǎng)查看一下自己的cuda對應(yīng)的版本了,我之前遇到過這情況,這么處理就可以了。
查看是否安裝完成的話,可以在該環(huán)境下輸入:文章來源:http://www.zghlxwxcb.cn/news/detail-429485.html
import mmdet3d
print(mmdet3d.__version__)
驗證
可以在jupyter lab里進行驗證。這里如果沒有jupyter lab的話,直接pip install jupyterlab
即可。
在命令行輸入jupyter lab進入jupyterlab。
輸入如下指令即可:
其實就是調(diào)用mmdet3d的api進行檢測,需要注意的是checkpoint需要自己去下載,可以在GitHub中的config目錄下下載對應(yīng)的model:
顯示結(jié)果如下:
如果是對點云的檢測就是這樣的:文章來源地址http://www.zghlxwxcb.cn/news/detail-429485.html
到了這里,關(guān)于MMdetection3D學(xué)習(xí)系列(一)——環(huán)境配置安裝的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!