OpenPCDet安装及其3D检测算法实现

这篇具有很好参考价值的文章主要介绍了OpenPCDet安装及其3D检测算法实现。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

前言

open-mmlab /OpenPCDet是一个简洁独立开源的适用于激光雷达3D目标检测的平台,支持多种基于深度学习的3D目标检测算法,可以使用开源数据集或者自制数据集进行训练测试。
github网址链接

一、配置环境

1.1 验证cmake 版本(3.13)

cmake --version

cmake官网链接
cmake版本要大于等于1.13,不然后边的库的setup过程会报错。不要看网上的博客盲目执行pip install cmake或者apt-get autoremove cmake,这样会导致出一些额外的错误,最好的方法是采用超链接的方法进行版本的修改。

1.2 验证cuda、cudnn版本(10.2)

nvcc -V和nvidia-smi两个命令显示的cuda是不一致的,这个问题不大。主要看nvcc -V这条命令显示的cuda版本。nvidia-smi所显示的cuda是,这个台机器安装驱动所能带动的最大cuda版本,nvcc -V显示的cuda版本才是运行是的版本。一般来说nvidia-smi所显示的版本要高一些。

1.3 确定python(3.7)、pytorch版本

安装pytorch时,安装命令,建议去官网复制粘贴,不容易出现cuda和torch版本不匹配的问题。有时也直接用下边的命令行,下载安装也非常快,比如 2.1的第三条指令。

二、安装spconv

2.1 创建虚拟环境

conda create -n pcdet python=3.7  # 创建python3.7环境
conda activate pcdet # 激活环境
conda install pytorch==1.5.0 torchvision==0.6.0 -c pytorch

验证安装成功

(pcdet) ypx@ypx:~/ws/OpenPCDet$ python
Python 3.7.13 (default, Mar 29 2022, 02:18:16) 
[GCC 7.5.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
>>> print(torch.__version__)
1.5.0
>>> print(torchvision.__version__)
0.6.0a0+82fd1c8
>>> 

openpcdet环境配置,3d,算法,pytorch
注:(1)pytorch与torchvision版本视具体工程算法、硬件要求进行适当选择。因为后期具体编译的需要,必要时需要提升版本。
(2)对于虚拟环境的pytorch与torchvision安装有conda install 与pip install两种。大部分情况下是没有差别,但是有些依赖会有问题,为保持一致性,最好选pip install。
(3)在虚拟环境中,注意pip install、pip3 install、conda install是有区别的:
https://www.zhihu.com/question/395145313
大部分情况下,二者没有区别,怎么安装都可以work,不过,二者混着装各种package你真的会因为版本冲突而发疯!conda有严格的检查机制,它会保证你当前装的package安装好之后能work,但是,它只检查用conda安装过的package。例如,你新安装的package会依赖numpy ,不过你已经安装numpy(e.g., 1.19.2),但是用pip安装的,不好意思,它会认为你没安装,然后用conda再安装一个依赖版本的numpy(e.g., 1.18.5)。这个时候,两个numpy可能就打架了。并且你pip list与conda list显示的numpy版本可能不一致,结果不知道最后运行程序的时候调用了哪个版本的numpy。这个时候,uninstall可能也卸载不干净,卸了一个还有一个,最后还可能一直list有却是一个空壳。这个时候,快刀斩乱麻的方法,找到anaconda的lib/site-packages/numpy, 手动删掉它!pip的一个好处是可以安装时既检查conda安装过package的也检查pip安装过的package。不过,它只负责要什么装什么,不负责能不能把装的一堆packages打通,可能装好不work。一般原则,在新环境中,如果装多个packages,既用到conda,又用到pip,那就先conda 的都装好,再pip,如果能用一种装到底,就不要来回换着用。

2.2 安装spconv

spconv是一个卷积库,分为spconv 1.x与spconv 2.x两种版本,具体可以参考:
OpenPCDet完整环境下Spconv1.x与Spconv2.x的安装问题及解决方法
spconv 的github链接

这一步比较容易出现问题。目前spconv出了2.0以上版本,而spconv2.只支持cuda10.2以上版本,如果cuda版本合适,直接运行pip安装即可,像我的系统安装的是cuda10.1,只能老老实实下载原始的版本。

2.2.1 找个文件夹,开始安装spconv1.2:

#安装libboost
sudo apt-get install libboost-all-dev
# 下载spconv,并切换分支
git clone -b v1.2.1 https://ghproxy.com/https://github.com/traveller59/spconv.git

openpcdet环境配置,3d,算法,pytorch

2.2.2 spconv项目中的third_party对应的目录中有三个文件夹是空的,删除了进行更换;

openpcdet环境配置,3d,算法,pytorch

cd third_party
git clone https://ghproxy.com/https://github.com/pybind/pybind11
#或者  git clone https://github.com/pybind/pybind11.git
cd pybind11
git checkout -b 3b1dbeb

openpcdet环境配置,3d,算法,pytorch

2.2.3 编译安装 spconv v1.2

cd .. #退回
cd ..
python3 setup.py bdist_wheel
#如setup.py这一步报错,可能需要你删掉build/ dist/ spconv.egg-info/,这几个文件夹之后再编译

成功编译后:
openpcdet环境配置,3d,算法,pytorch

注:这一步极有可能报错,

报错一:

openpcdet环境配置,3d,算法,pytorch
openpcdet环境配置,3d,算法,pytorch

报错二:
openpcdet环境配置,3d,算法,pytorch
此报错,什么都不需要修改,直接再执行一次。
报错三:
openpcdet环境配置,3d,算法,pytorch

openpcdet环境配置,3d,算法,pytorch

解决办法参考:

运行setup.py无法编译spconv的问题–ubuntu18.04

ubuntu18.04安装spconv踩坑实录

cout不是std的成员

(1)当无具体对应的报错解决方案时,需要针对自己的具体报错,进行百度搜索

(2)若出现报错nvcc fatal: unknown ‘-Wall’ 注释文件 中包含 -Wall 的两行代码 并重新执行
Caffe2Targets.cmake 文件路径如下:

~/anaconda3/envs/pcdet/lib/python3.6/site-packages/torch/share/cmake/Caffe2/Caffe2Targets.cmake

两行代码如下:

(line 73) INTERFACE_COMPILE_OPTIONS "-Wall;-Wextra;-Wno-unused-parameter;-Wno-missing-field-initializers;-Wno-write-strings;-Wno-unknown-pragmas;-Wno-missing-braces;-fopenmp"

(line 93) INTERFACE_COMPILE_OPTIONS "-Wall;-Wextra;-Wno-unused-parameter;-Wno-missing-field-initializers;-Wno-write-strings;-Wno-unknown-pragmas;-Wno-missing-braces"

2.2.4 安装:

cd ./dist
pip install spconv-1.2.1-cp37-cp37m-linux_x86_64.whl# 先检查一下dist里边的文件,有可能是空的

openpcdet环境配置,3d,算法,pytorch

2.2.5 验证是否安装成功

python
import spconv

结果无报错即可

三、安装Openpcdet

git clone https://ghproxy.com/https://github.com/open-mmlab/OpenPCDet.git
cd OpenPCDet
pip install -r requirements.txt 
python setup.py develop 

openpcdet环境配置,3d,算法,pytorch
要确认是安装的完整版的pcdet,如果不是,重新编译,或者更换pytorch版本编译。
openpcdet环境配置,3d,算法,pytorch

验证pcdet是否安装成功,无报错即可功,:

python 
import pcdet

参考链接:
安装 open-mmlab /OpenPCDet并测试demo

三维目标检测之OpenPCDet环境配置及demo测试

openpcdet环境配置及demo运行

四、复现demo(以pointpillar算法为例)

4.1 快速实现demo

4.1.1下载pointpillar的预训练模型

PointPillars点云检测在OpenPCDet推理代码详解

若复现目标是其他算法,则选择对应的预训练模型
下载链接pointpillar预训练模型
openpcdet环境配置,3d,算法,pytorch

将下载好的文件pointpillar_7728.pth 放入文件夹/OpenPCDet/tools

4.1.2 下载kitti数据集

kitti官网下载链接
object-> 3d object

选择的是激光雷达的3D点云数据:
Download Velodyne point clouds, if you want to use laser information (29 GB)
下载完成后解压发现里面有两个文件夹,分为训练集和测试集:training和testing
下载下来后将其放置一个固定位置。

openpcdet环境配置,3d,算法,pytorch

4.2 自己训练模型实现demo(以pointpillar_pyramid_aug为例)

4.2.1 准备数据集(从kitti下载)

openpcdet环境配置,3d,算法,pytorch

4.2.2 进行数据处理

python -m pcdet.datasets.kitti.kitti_dataset create_kitti_infos tools/cfgs/dataset_configs/kitti_dataset.yaml

这一命令用来生成info文件:

openpcdet环境配置,3d,算法,pytorchopenpcdet环境配置,3d,算法,pytorch

4.2.3 进行训练

python train.py --cfg_file ./cfgs/kitti_models/pointpillar_pyramid_aug.yaml  --batch_size 4 --epochs 50

openpcdet环境配置,3d,算法,pytorch
openpcdet环境配置,3d,算法,pytorch
训练完后,终端末尾会自动生成准确度的评估数据,自行保存:

2023-05-21 03:25:04,995   INFO  Car AP@0.70, 0.70, 0.70:
bbox AP:90.7175, 89.4936, 88.7064
bev  AP:89.2734, 87.1144, 85.1177
3d   AP:86.5935, 76.9714, 75.3080
aos  AP:90.67, 89.22, 88.31
Car AP_R40@0.70, 0.70, 0.70:
bbox AP:95.4567, 91.8373, 91.0719
bev  AP:91.5083, 87.6349, 86.6663
3d   AP:87.2124, 78.0541, 75.3106
aos  AP:95.40, 91.54, 90.63
Car AP@0.70, 0.50, 0.50:
bbox AP:90.7175, 89.4936, 88.7064
bev  AP:94.0103, 89.9321, 89.3273
3d   AP:93.9647, 89.8485, 89.1614
aos  AP:90.67, 89.22, 88.31
Car AP_R40@0.70, 0.50, 0.50:
bbox AP:95.4567, 91.8373, 91.0719
bev  AP:96.9474, 94.3945, 93.7736
3d   AP:96.6617, 94.2059, 93.3433
aos  AP:95.40, 91.54, 90.63
Pedestrian AP@0.50, 0.50, 0.50:
bbox AP:64.1480, 59.0424, 56.4814
bev  AP:59.1269, 53.9681, 50.1294
3d   AP:54.8211, 49.2334, 46.0115
aos  AP:46.23, 42.94, 40.65
Pedestrian AP_R40@0.50, 0.50, 0.50:
bbox AP:63.8036, 59.0105, 55.9399
bev  AP:58.6899, 52.9762, 48.8099
3d   AP:54.1183, 48.2686, 44.0606
aos  AP:43.62, 40.54, 37.97
Pedestrian AP@0.50, 0.25, 0.25:
bbox AP:64.1480, 59.0424, 56.4814
bev  AP:68.7356, 65.3176, 62.6414
3d   AP:68.6939, 65.1772, 62.2696
aos  AP:46.23, 42.94, 40.65
Pedestrian AP_R40@0.50, 0.25, 0.25:
bbox AP:63.8036, 59.0105, 55.9399
bev  AP:70.0396, 65.7866, 62.6072
3d   AP:69.9971, 65.5463, 62.3042
aos  AP:43.62, 40.54, 37.97
Cyclist AP@0.50, 0.50, 0.50:
bbox AP:87.1074, 78.0943, 74.0882
bev  AP:82.9921, 67.9041, 63.8165
3d   AP:81.2777, 64.7565, 60.6614
aos  AP:86.04, 75.29, 71.34
Cyclist AP_R40@0.50, 0.50, 0.50:
bbox AP:90.2654, 79.5607, 75.1355
bev  AP:86.0327, 68.1021, 63.9980
3d   AP:83.8334, 64.8156, 60.6342
aos  AP:89.04, 76.44, 71.93
Cyclist AP@0.50, 0.25, 0.25:
bbox AP:87.1074, 78.0943, 74.0882
bev  AP:85.3837, 74.4590, 70.6709
3d   AP:85.3837, 74.4590, 70.6709
aos  AP:86.04, 75.29, 71.34
Cyclist AP_R40@0.50, 0.25, 0.25:
bbox AP:90.2654, 79.5607, 75.1355
bev  AP:88.6348, 75.5956, 71.3170
3d   AP:88.6348, 75.5956, 71.3167
aos  AP:89.04, 76.44, 71.93

训练模型的保存路径:
openpcdet环境配置,3d,算法,pytorch

4.2.4 查看损失函数

pip install tensorboardX
pip install tensorflow

然后进行到events.out.tfevents.1684592165.ypx所在的路经/home/ypx/ws/OpenPCDet/OpenPCDet/output/cfgs/kitti_models/pointpillar_pyramid_aug/default/tensorboard执行命令:

tensorboard --logdir=/home/ypx/ws/OpenPCDet/OpenPCDet/output/cfgs/kitti_models/pointpillar_pyramid_aug/default/tensorboard

4.3 选择并下载可视化工具

可视化工具有两款可以选择:mayavi与open3d
至于尝试进行ros下的可视化,可以参考另一篇博客:三维目标检测之ROS可视化

4.3.1 选择open3d进行可视化

open3d的版本选择跟ubuntu的版本号也有关系,这个要注意。具体可以参考这篇博客:
Open3D版本和Ubuntu的关系

(1) 安装指令:

conda install -c open3d-admin open3d==0.9.0
pip install open3d-python

open3d==0.9.0的安装
openpcdet环境配置,3d,算法,pytorch
openpcdet环境配置,3d,算法,pytorch

open3d-python的安装

openpcdet环境配置,3d,算法,pytorchopenpcdet环境配置,3d,算法,pytorch

(2) 验证open3d是否安装成功:

python
import open3d

openpcdet环境配置,3d,算法,pytorch

(3)指令运行

python3 -s  demo.py --cfg_file cfgs/kitti_models/pointpillar.yaml  --ckpt pointpillar_7728.pth --data_path /home/ypx/ws/OpenPCDet/data_object_velodyne/testing/velodyne/000000.bin

如果是自己训练的模型,则:

python demo.py --cfg_file cfgs/kitti_models/pointpillar_pyramid_aug.yaml  --ckpt /home/ypx/ws/OpenPCDet/OpenPCDet/output/cfgs/kitti_models/pointpillar_pyramid_aug/default/ckpt/checkpoint_epoch_50.pth --data_path /home/ypx/ws/OpenPCDet/OpenPCDet/data/kitti/testing/velodyne/000517.bin

其中:

–cfg_file: .yaml配置文件
–ckpt:  为训练好的.path 模型文件
–data_path:为检测的数据.bin文件
-s:  有时候之前安装过open3d,所以在~/.local中有一个版本错误的open3d,在运行demo程序的时候,需要加上-s,-s可以让Python不去查找用户目录中的库。

(4)如何想用GPU运行模型,将demo.py的84行语句进行修改:

model.load_params_from_file(filename=args.ckpt, logger=logger, to_cpu=True)

改为:

model.load_params_from_file(filename=args.ckpt, logger=logger, to_cpu=False)

(5)解决报错
有时执行完这条语句后,会出现报错

报错一:
openpcdet环境配置,3d,算法,pytorch
报错二:

AttributeError: 'open3d.geometry.PointCloud()' object has no attribute 'voxel_down_sample'

报错三:

ImportError: /lib/x86_64-linux-gnu/libm.so.6: version `GLIBC_2.27' not found (required by /home/xxxx/xxxx/lib/python3.7/site-packages/open3d/open3d_pybind.cpython-37m-x86_64-linux-gnu.so)

以上报错,
这大概率是open3d版本与ubuntu不对应,或者安装了多个版本的open3d,此时可以删除重新安装:

conda remove open3d

重新安装时,最好4.3.1的(1)的两条指令一起执行。
openpcdet环境配置,3d,算法,pytorch

报错四:
openpcdet环境配置,3d,算法,pytorch
openpcdet环境配置,3d,算法,pytorch
open3d有时候会缺少一些部分功能包,可以尝试 pip install <功能包名>;
但是注意安装位置,有时是./local,而不是当前新建的虚拟环境:这时要么想办法调用./local的文件(去掉执行命令中的-s,python3换为python),要么想办法确认当前的python版本(pip3)和安装指令,安装到当前环境中

openpcdet环境配置,3d,算法,pytorch

报错五:
段错误(核心已转储)
openpcdet环境配置,3d,算法,pytorch
或者

(pcdet) auto@auto-System-Product-Name:~/OpenPCDet/tools$ python demo.py --cfg_file cfgs/kitti_models/pv_rcnn.yaml --ckpt pv_rcnn_8369.pth --data_path ../data/kitti/testing/velodyne/000008.bin 
2022-03-06 21:32:26,628   INFO  -----------------Quick Demo of OpenPCDet-------------------------
2022-03-06 21:32:26,628   INFO  Total number of samples:        1
/home/auto/anaconda3/envs/pcdet/lib/python3.7/site-packages/torch/functional.py:445: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at  ../aten/src/ATen/native/TensorShape.cpp:2157.)
  return _VF.meshgrid(tensors, **kwargs)  # type: ignore[attr-defined]
2022-03-06 21:32:29,122   INFO  ==> Loading parameters from checkpoint pv_rcnn_8369.pth to CPU
2022-03-06 21:32:29,154   INFO  ==> Done (loaded 367/367)
2022-03-06 21:32:29,300   INFO  Visualized sample index:        1
段错误 (核心已转储)

解决办法参考,【段错误(核心已转储)——运行PV-RCNN代码 python demo.py】

报错六:
openpcdet环境配置,3d,算法,pytorch
pytorch版本过低,某些函数无法激活,更换版本。

(4)可视化结果:
openpcdet环境配置,3d,算法,pytorch

4.3.2 选择mayavi进行可视化

(1) tools/demo.py文件中,优先使用的是open3d,如下图所示,如果想要用mayavi这个库的话,需要把try部分删了,只用except里边的部分。
原文件:

import argparse
import glob
from pathlib import Path

try:
    import open3d
    from visual_utils import open3d_vis_utils as V
    OPEN3D_FLAG = True
except:
    import mayavi.mlab as mlab
    from visual_utils import visualize_utils as V
    OPEN3D_FLAG = False

import numpy as np
import torch

from pcdet.config import cfg, cfg_from_yaml_file
from pcdet.datasets import DatasetTemplate
from pcdet.models import build_network, load_data_to_gpu
from pcdet.utils import common_utils

将这一块去除就可以了:

try:
    import open3d
    from visual_utils import open3d_vis_utils as V
    OPEN3D_FLAG = True
except:

修改后:

import argparse
import glob
from pathlib import Path
import mayavi.mlab as mlab
from visual_utils import visualize_utils as V
OPEN3D_FLAG = False

import numpy as np
import torch

from pcdet.config import cfg, cfg_from_yaml_file
from pcdet.datasets import DatasetTemplate
from pcdet.models import build_network, load_data_to_gpu
from pcdet.utils import common_utils

(2)安装

pip install vtk==8.1.2 # 自动安装的vtk是9,会产生冲突
pip install mayavi==4.7.3
pip install PyQt5
pip install scipy
pip install scikit-image
pip install tqdm
pip install torchvision

(3)报错解决:
python3.8安装mayavi4.7.3
Conda安装mayavi
python安装mayavi后报错AttributeError

(4)指令运行

python demo.py --cfg_file cfgs/kitti_models/pointpillar.yaml     --ckpt pointpillar_7728.pth     --data_path /home/ypx/ws/OpenPCDet/data_object_velodyne/testing/velodyne/000000.bin

整体参考:
ubuntu 18.04 从零开始复现 OpenPCDet 训练kitti数据集并评估 可视化 损失函数
OpenPCDet——环境配置和训练测试(升级系统、Nvidia驱动、cuda11.3、cudnn8.2)
OpenPCDet详细安装指南+单/多GPU训练演示
Wait 30 seconds for next check
Openpcdet训练自己的数据集
【OpenPCDet】Kitti数据集下训练PointPillars并评估&可视化
基于OpenPCDet实现自定义数据集的训练,狸花猫看完要打拳系列(一)!
OpenPCDet 踩坑实录
OpenPCDet 训练KITTI

KITTI 3D目标检测的评估指标
深度学习算法评价标准AP bbox bev 3d

可视化参考:
PointPillars与PointRCNN编译运行以及结果可视化(小白探索之路,清晰易懂)

点云 3D 物体检测结果(SECOND、Pointpillars、PointRCNN)的几种可视化方法

PointRCNN测试+可视化完整流程

pointRCNN 结果可视化

Open3d对kitti数据集可视化

【open3d】显示kitti 点云数据和bbox文章来源地址https://www.toymoban.com/news/detail-784450.html

到了这里,关于OpenPCDet安装及其3D检测算法实现的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • OpenPCDet安装、使用方式及自定义数据集训练

    个人博客 安装完,就可以运行 demo.py ,测试一下。(需要准备好模型和数据文件) 如果出现 SharedArray 相关的错误的话,可以适当的降低其版本。例如 pip install -U SharedArray==3.1 注意, demo.py 运行成功需要在具有显示设备的条件下,如果只有终端的话是无法运行成功的。 首先需

    2024年02月09日
    浏览(38)
  • Ubuntu 22.04从零开始安装Openpcdet 0.6.0——详细记录过程(含4090+CUDA11.8)

    A、云平台安装和本地安装的差异: 之前在几个 云平台 上的1w算力豆已经用光了,所以这次本地选择 本地安装Pcdet 。 两者安装过程 有很大差异,因为 九天毕昇云平台 使用的是CUDA10.1,且没办法改动,导致pcdet 只能安装0.3.2,且只能选择较低的版本的第三方库:torch1.6.0,sp

    2024年02月16日
    浏览(46)
  • 基于OpenPCDet实现自定义数据集的训练,狸花猫看完要打拳系列(一)!

      最近在学习如何基于 OpenPCDet框架进行PointPillars网络训练 ,由于对框架以及完整训练过程都不了解,因此打算记录下自己的学习过程,感谢学习过程中狸花猫sensei的大力支持,目标是实现自定义数据集(因为笔者 只有激光雷达的数据,仿照kitti格式进行标注 )的训练,然后

    2024年02月09日
    浏览(35)
  • OpenPCDet解析

    OpenPCDet: Open-MMLab 面向LiDAR点云表征的3D目标检测代码库 OpenPCDet的github链接 OpenPCDet 是一套基于 PyTorch 实现的 点云3D目标检测代码库 。 设计思想 : 点云数据集 (KITTI、NuScene、Lyft、Waymo、PandaSet等)在数据格式与3D坐标系上往往定义各不相同,各式各样的 点云感知算法 (point

    2024年02月04日
    浏览(37)
  • Openpcdet训练自己的数据集

    * Openpcdet详细内容请看以下链接: GitHub - open-mmlab/OpenPCDet: OpenPCDet Toolbox for LiDAR-based 3D Object Detection. 1.首先gitclone原文代码 2. 这里我建议自己按照作者github上的docs/install文件夹下指示一步步安装,(之前根据csdn上教程一直有报错),然后下载spconv,以及cumm, github链接如下: GitH

    2024年03月24日
    浏览(36)
  • PointPillars论文解析和OpenPCDet代码解析

            PointPillars是一个来自工业界的模型,整体思想基于图片的处理框架,直接将点云从俯视图的视角划分为一个个的Pillar(立方柱体),从而构成了类似 图片 的数据,然后在使用2D的检测框架进行特征提取和密集的框预测得到检测框,从而使得该模型在速度和精度都

    2024年01月19日
    浏览(42)
  • Openpcdet 系列 Pointpillar代码逐行解析之POST_PROCESSING模块

    该模块主要负责对检测结果进行滤波、聚类、追踪等操作,以提高检测的准确性和稳定性。 POST_PROCESSING模块通常包含以下几个主要的子模块或步骤: 点云滤波(Point Cloud Filtering):这一步骤用于去除原始点云中的噪声和离群点,常用的滤波方法包括体素下采样(Voxel Downsam

    2024年02月03日
    浏览(58)
  • OpenPCDet系列 | 7.1 KITTI数据集测试流程predicted_boxes预测

    测试流程的结构图如下所示: generate_predicted_boxes函数 一开始的数据传入为: 首先对于各类预测的特征图进行重新reshape处理,将anchor那一维度进行拼接操作,比如:(16, 248, 216, 42) - (16, 321408, 7)。但是这里需要注意。 特征预测的box信息是基于anchor的一个偏移,也就是编码后的

    2024年02月06日
    浏览(37)
  • 无人驾驶实战-第五课(动态环境感知与3D检测算法)

    在七月算法上报了《无人驾驶实战》课程,老师讲的真好。好记性不如烂笔头,记录一下学习内容。 课程入口,感兴趣的也可以跟着学一下。 ————————————————————————————————————————— 激光雷达的分类:     机械式Lidar:

    2024年02月14日
    浏览(38)
  • 视频异常检测数据集 (ShanghaiTech) 及其I3D特征转换

    ShanghaiTech 是一个中型数据集,基本信息如下: 训练集 :330个正常视频; 测试集 :107个异常视频,已被划分为多个帧,包含13个异常事件,且带有。 该数据集的一个示意如下图: 为了使得数据集适应MIL的场景,Zhong等人依据类别平衡的准则将整个数据集重新划分。划分的索

    2023年04月18日
    浏览(76)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包