Skip to content

Latest commit

 

History

History
141 lines (72 loc) · 6.9 KB

model_zoo.md

File metadata and controls

141 lines (72 loc) · 6.9 KB

模型库

通用设置

  • 使用分布式训练;
  • 为了和其他代码库做公平对比,本文展示的是使用 torch.cuda.max_memory_allocated() 在 8 个 GPUs 上得到的最大 GPU 显存占用值,需要注意的是,这些显存占用值通常小于 nvidia-smi 显示出来的显存占用值;
  • 在模型库中所展示的推理时间是包括网络前向传播和后处理所需的总时间,不包括数据加载所需的时间,模型库中所展示的结果均由 benchmark.py 脚本文件在 2000 张图像上所计算的平均时间。

基准结果

SECOND

请参考 SECOND 获取更多的细节,我们在 KITTI 和 Waymo 数据集上都给出了相应的基准结果。

PointPillars

请参考 PointPillars 获取更多细节,我们在 KITTI 、nuScenes 、Lyft 、Waymo 数据集上给出了相应的基准结果。

Part-A2

请参考 Part-A2 获取更多细节。

VoteNet

请参考 VoteNet 获取更多细节,我们在 ScanNet 和 SUNRGBD 数据集上给出了相应的基准结果。

Dynamic Voxelization

请参考 Dynamic Voxelization 获取更多细节。

MVXNet

请参考 MVXNet 获取更多细节。

RegNetX

请参考 RegNet 获取更多细节,我们将 pointpillars 的主干网络替换成 RegNetX,并在 nuScenes 和 Lyft 数据集上给出了相应的基准结果。

nuImages

我们在 nuImages 数据集 上也提供基准模型,请参考 nuImages 获取更多细节,我们在该数据集上提供 Mask R-CNN , Cascade Mask R-CNN 和 HTC 的结果。

H3DNet

请参考 H3DNet 获取更多细节。

3DSSD

请参考 3DSSD 获取更多细节。

CenterPoint

请参考 CenterPoint 获取更多细节。

SSN

请参考 SSN 获取更多细节,我们将 pointpillars 中的检测头替换成 SSN 模型中所使用的 ‘shape-aware grouping heads’,并在 nuScenes 和 Lyft 数据集上给出了相应的基准结果。

ImVoteNet

请参考 ImVoteNet 获取更多细节,我们在 SUNRGBD 数据集上给出了相应的结果。

FCOS3D

请参考 FCOS3D 获取更多细节,我们在 nuScenes 数据集上给出了相应的结果。

PointNet++

请参考 PointNet++ 获取更多细节,我们在 ScanNet 和 S3DIS 数据集上给出了相应的结果。

Group-Free-3D

请参考 Group-Free-3D 获取更多细节,我们在 ScanNet 数据集上给出了相应的结果。

ImVoxelNet

请参考 ImVoxelNet 获取更多细节,我们在 KITTI 数据集上给出了相应的结果。

PAConv

请参考 PAConv 获取更多细节,我们在 S3DIS 数据集上给出了相应的结果。

DGCNN

请参考 DGCNN 获取更多细节,我们在 S3DIS 数据集上给出了相应的结果。

SMOKE

请参考 SMOKE 获取更多细节,我们在 KITTI 数据集上给出了相应的结果。

PGD

请参考 PGD 获取更多细节,我们在 KITTI 和 nuScenes 数据集上给出了相应的结果。

PointRCNN

请参考 PointRCNN 获取更多细节,我们在 KITTI 数据集上给出了相应的结果。

MonoFlex

请参考 MonoFlex 获取更多细节,我们在 KITTI 数据集上给出了相应的结果。

SA-SSD

请参考 SA-SSD 获取更多的细节,我们在 KITTI 数据集上给出了相应的基准结果。

FCAF3D

请参考 FCAF3D 获取更多的细节,我们在 ScanNet, S3DIS 和 SUN RGB-D 数据集上给出了相应的基准结果。

PV-RCNN

请参考 PV-RCNN 获取更多的细节,我们在 KITTI 数据集上给出了相应的基准结果。

BEVFusion

请参考 BEVFusion 获取更多的细节, 我们在 NuScenes 数据集上给出了相应的基准结果。

CenterFormer

请参考 CenterFormer 获取更多的细节, 我们在 Waymo 数据集上给出了相应的基准结果。

TR3D

请参考 TR3D 获取更多的细节, 我们在 ScanNet, SUN RGB-D 和 S3DIS 数据集上给出了相应的基准结果。

DETR3D

请参考 DETR3D 获取更多的细节, 我们在 NuScenes 数据集上给出了相应的基准结果。

PETR

请参考 PETR 获取更多的细节, 我们在 NuScenes 数据集上给出了相应的基准结果。

TPVFormer

请参考 TPVFormer 获取更多的细节, 我们在 NuScenes 数据集上给出了相应的基准结果。

Mixed Precision (FP16) Training

细节请参考 Mixed Precision (FP16) Training 在 PointPillars 训练的样例