AB3DMOT 车辆 行人识别

代码库:https://github.com/xinshuoweng/AB3DMOT

搭建环境:
Ubuntu 16.04、Python 3.7.6 in Anaconda、Pytorch 1.2 + CUDA 9.2
预研环境搭建参照PointRCNN环境搭建及运行所述

KITTI object tracking数据集下载

数据集官网:http://www.cvlibs.net/datasets/kitti/eval_tracking.php
按照项目说明进行预研时,只需要下载image_02数据集:
http://www.cvlibs.net/download.php?file=data_tracking_image_2.zip

Xinshuo_PyToolbox准备

该项目需要作者的工具包https://github.com/xinshuoweng/Xinshuo_PyToolbox
需要安装其requirements.txt所述的package,之后将其中所有文件夹直接放入AB3DMOT文件夹内即可

步骤


  1. install dependeny:
$ cd path/to/AB3DMOT
$ pip install -r requirements.txt

  2. Inference - run our tracker on the KITTI MOT validation set with the provided detection:
$ python main.py pointrcnn_Car_val
$ python main.py pointrcnn_Pedestrian_val
$ python main.py pointrcnn_Cyclist_val
选择一条即可,例如pedestrian

  3. 将image_2数据包内的文件解压至data/KITTI/resources

  4. 修改visualization.py内的变量seq_list为想要可视化的sample number,例如:
seq_list = ['0000', '0001', '0002', '0003', '0004', '0005', '0006', '0007', '0008'] 

  5. 执行python visualization.py pointrcnn_Pedestrian_val (对应步骤2的pointrcnn_Pedestrian_val)

  6. 在results/pointrcnn_Pedestrian_val/trk_image_vis查看结果

效果评价

  • 容易将长方体/柱状物识别成车辆/行人,识别车辆时尤为明显,但这种误判一般发生在道路区域外,-且持续帧数很少;
  • 可能将骑自行车的人判别为一般行人;
  • 没有明显漏判情况,但在发生遮挡时追踪很可能被打断致使追踪对象id不稳定;
  • 2
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值