1. 程式人生 > >機器人視覺專案:視覺檢測識別+機器人跟隨(2)

機器人視覺專案:視覺檢測識別+機器人跟隨(2)

目標:在TX2上執行人體檢測和跟蹤演算法

主體方案:人體檢測[yolov3] + 跟蹤[KCF]

現狀

  1. yolov3在tx2的幀率是2fps,yolov3-tiny能達到15~20fps。

  2. KCF很快,但是很容易跟丟。

  3. 目前用的是RGB影象,如果用RGBD,是不是會更好?

幾個可能改進的方向

  1. 用TensorRT跑yolov3,提升yolov3在tx2上的速率。

  2. 如果可能,用機場環境,自己標註資料,訓練模型。

  3. 針對人體這類特殊的目標,改進KCF所使用的特徵。

  4. 使用前2幀預測目標運動方向,改進KCF的搜尋範圍。

  5. 用GPU版本的KCF實現。