摘要:針對機械臂藥盒抓取操作中對藥盒定位和姿態(tài)估計的要求,提出一種基于YOLOv3深度學(xué)習算法和EPnP算法相結合的多藥盒姿態(tài)估計方法,此方法主要分為多藥盒定位和姿態(tài)估計兩部分。首先通過(guò)YOLOv3算法實(shí)現藥盒的快速精確定位,并通過(guò)定位框分割出單個(gè)藥盒。然后進(jìn)行特征提取和特征匹配并估計單應矩陣。通過(guò)對單應矩陣的透視矩陣變換求得藥盒平面四個(gè)角點(diǎn)的像素坐標并作為EPnP求解所需的2D點(diǎn),結合藥盒先驗尺寸信息在相機坐標系下構建藥盒對應的3D點(diǎn)坐標以實(shí)現藥盒姿態(tài)求解。通過(guò)結合OptiTrack系統設計了藥盒姿態(tài)精度對比實(shí)驗,結果表明,本算法充分發(fā)揮了YOLOv3算法兼具快速性和準確性的優(yōu)勢,并且具有良好的姿態(tài)估計精度,總體算法速度達到15FPS,藥盒姿態(tài)估計平均誤差小于0.5度。