隨著自主機器的發展,我們可以在生活中經常看到自主機器的應用。有傳統應用的倉庫工廠AMR,機械臂,銀行酒店里面的服務機器人,家庭機器人,無人物流車,自主礦卡等等。不同的自主機器,軟件架構的方案也不一樣,但核心的模塊定位,導航,感知,控制等都是相通的。
定位模塊是自主機器最核心的模塊之一,定位又包括全局定位和局部定位,對于自主機器,其精度需要達到厘米級別。本文我們將討論全局定位,即確定自主機器在全局下的位置。傳統的低速自主機器,類似于AMR等,其采用的定位方式通常以SLAM(simultaneous localization and mapping)的方法進行同時建圖和定位,但是該方法實現代價高,難度大,并不適用于室外自主機器,類似于無人物流車,園區接駁車等的實時高精度定位需求。這些室外自主機器行駛速度快,距離遠,環境復雜,使得SLAM的精度下降,同時遠距離的行駛將導致實時構建的地圖偏移過大。因此,如果在已有高精度的全局地圖地圖的情況下進行自主機器的定位,將極大的簡化該問題。
因此,將問題分為獨立的兩部分:建圖Mapping和定位Matching。NDT是一種點云配準算法,可同時用于點云的建圖和定位。
CUDA-NDT
正態分布變換算法(Normal Distributions Transform, NDT) 同ICP算法的功能一致,即,用于計算兩幀點云數據之間的坐標變換矩陣,從而能夠使不同的坐標下的點云數據合并到同一個坐標系統中。不同的是NDT算法對初值不敏感,且不需要進行對應點的特征計算,所以速度較快。NDT算法使用應用于 3D 點統計模型的標準優化技術來確定兩個點云之間最可能的配準。NDT算法和 ICP算法可以結合使用,以提高配準精度和速度。首先,NDT算法可用于粗配準,得到轉換參數;然后使用ICP算法結合參數進行精細配準。為了改進NDT算法在NVIDIA Jetson上的性能,我們推薦使用基于CUDA加速的CUDA-NDT。
使用CUDA-NDT
以下是CUDA NDT的使用實例
我們需要初始化相關的類對象,設置相關的參數,并調用接口函數。
cudaNDT ndtTest(nPCountM, nQCountM, stream);
ndtTest.setInputSource(source);
ndtTest.setInputTarget(target);
ndtTest.setResolution(resolution);
ndtTest.setMaximumIterations(nr_iterations);
ndtTest.setTransformationEpsilon(epsilon);
ndtTest.setStepSize(step_size);
ndtTest.ndt(cloud_source, nPCount,
cloud_target, nQCount, guess,
transformation_matrix, stream);
CUDA-NDT 計算的輸出是 transformation_matrix,代表的含義如下:
源點云(P)* transformation_matrix = 目標坐標系的點云(Q)
因為激光類型的輸出點云的數量為固定值,所以CUDA-NDT在輸出化的時候,要求輸入兩幀點云的最大數量,從而分配計算資源。
class cudaNDT
{
public:
/*
nPCountM and nQCountM are the maximum of count for input clouds
They are used to pre-allocate memory.
*/
cudaNDT(int nPCountM, int nQCountM, cudaStream_t stream = 0);
~cudaNDT(void);
void setInputSource (void *source);
void setInpuTarget (void *target);
void setResolution (float resolution);
void setMaximumIterations (int nr_iterations);
void setTransformationEpsilon (double epsilon);
void setStepSize (double step_size);
/*
cloud_target = transformation_matrix * cloud_source
When the Epsilon of transformation_matrix is less than threshold,
the function will return transformation_matrix.
Input:
cloud_source, cloud_target: data pointer for points cloud
nPCount: the points number of cloud_source
nQCount: the points number of cloud_target
guess: initial guess of transformation_matrix
stream: CUDA stream
Output:
transformation_matrix: rigid transformation matrix
*/
void ndt(float *cloud_source, int nPCount,
float *cloud_target, int nQCount,
float *guess, void *transformation_matrix,
cudaStream_t stream = 0);
void *m_handle = NULL;
};
經過CUDA加速的NDT速度對比微加速版本提升了4倍左右,請參考以下Table2性能對比,經過NDT匹配的點云效果對比請參考Figure1和2。
關于作者
Lily Li 正在為 NVIDIA 的機器人團隊處理開發人員關系。她目前正在 Jetson 生態系統中開發機器人技術解決方案,以幫助創建最佳實踐。
Haoyu Deng 是 NVIDIA 的CUDA開發工程師。目前,他正與 TSE 中國團隊合作,通過CUDA開發優化軟件性能的解決方案。
審核編輯:郭婷
-
機器人
+關注
關注
211文章
28620瀏覽量
207929 -
AMR
+關注
關注
3文章
436瀏覽量
30302
發布評論請先 登錄
相關推薦
評論