TX2 使用板载摄像头进行KITTI车辆检测
KITTI数据集
KITTI数据集由德国卡尔斯鲁厄理工学院和丰田美国技术研究院联合创办,是目前国际上最大的自动驾驶场景下的计算机视觉算法评测数据集。有兴趣进一步了解的可以阅读这篇博客(感觉蛮不错的,虽然我没看完),还可以进入KITTI官网自行探索。
环境搭建
我也是参照网上的各种教程搭建的,而且时间有点久远了。如果有时间再补上环境搭建过程。
环境(版本不同大部分情况不影响):
- cuda 9.0
- opencv 4.0-pre
- nvcaffe 0.15.14
- digits 6.1.1
训练模型
按照DIGITS的github官网目标检测教程进行操作,教程难度不大,但有几个地方需要注意以下,如果不想自己训练的可以下载我训练好的模型,并直接python detectnet-camera.py
运行。