├── .gitignore ├── CMakeLists.txt ├── README.md ├── cmake └── utils.cmake ├── config ├── iassd_hvcsx2_4x8_80e_kitti_3cls(export).cache ├── iassd_hvcsx2_4x8_80e_kitti_3cls(export).onnx ├── iassd_hvcsx2_gq_4x8_80e_kitti_3cls(export).onnx └── trt.yaml ├── doc ├── check.md ├── demo.gif ├── gridballquery.md ├── img.png └── profile.md ├── inc ├── buffer.h ├── calibrator.h ├── helper.h ├── model.h └── visualization.h ├── package.xml ├── plugins ├── CMakeLists.txt ├── README.md └── src │ ├── CMakeLists.txt │ ├── FPSamplingPlugin │ ├── CMakeLists.txt │ ├── FPSampling.cu │ ├── FPSampling.h │ ├── FPSamplingPlugin.cpp │ ├── FPSamplingPlugin.h │ └── README.md │ ├── ballQueryPlugin │ ├── CMakeLists.txt │ ├── README.md │ ├── ballQueryPlugin.cu │ └── ballQueryPlugin.h │ ├── common │ ├── CMakeLists.txt │ ├── checkMacrosPlugin.cpp │ ├── checkMacrosPlugin.h │ ├── common.cu │ ├── common.h │ ├── hash.h │ ├── plugin.h │ ├── plugin_auto_declare.h │ └── print.h │ ├── exports.map │ ├── gridBallQueryPlugin │ ├── CMakeLists.txt │ ├── README.md │ ├── gridBallQueryPlugin.cu │ └── gridBallQueryPlugin.h │ ├── havSamplingForQueryPlugin │ ├── CMakeLists.txt │ ├── README.md │ ├── havSamplingQ.cu │ ├── havSamplingQ.h │ ├── havSamplingQPlugin.cu │ └── havSamplingQPlugin.h │ ├── havSamplingPlugin │ ├── CMakeLists.txt │ ├── README.md │ ├── havSampling.cu │ ├── havSampling.h │ ├── havSamplingPlugin.cu │ └── havSamplingPlugin.h │ └── nms3DPlugin │ ├── CMakeLists.txt │ ├── README.md │ ├── nms3D.cu │ ├── nms3D.h │ ├── nms3DPlugin.cpp │ └── nms3DPlugin.h └── src ├── pcvt.py └── point_detector.cpp /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/.gitignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/README.md -------------------------------------------------------------------------------- /cmake/utils.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/cmake/utils.cmake -------------------------------------------------------------------------------- /config/iassd_hvcsx2_4x8_80e_kitti_3cls(export).cache: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/config/iassd_hvcsx2_4x8_80e_kitti_3cls(export).cache -------------------------------------------------------------------------------- /config/iassd_hvcsx2_4x8_80e_kitti_3cls(export).onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/config/iassd_hvcsx2_4x8_80e_kitti_3cls(export).onnx -------------------------------------------------------------------------------- /config/iassd_hvcsx2_gq_4x8_80e_kitti_3cls(export).onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/config/iassd_hvcsx2_gq_4x8_80e_kitti_3cls(export).onnx -------------------------------------------------------------------------------- /config/trt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/config/trt.yaml -------------------------------------------------------------------------------- /doc/check.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/doc/check.md -------------------------------------------------------------------------------- /doc/demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/doc/demo.gif -------------------------------------------------------------------------------- /doc/gridballquery.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/doc/gridballquery.md -------------------------------------------------------------------------------- /doc/img.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/doc/img.png -------------------------------------------------------------------------------- /doc/profile.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/doc/profile.md -------------------------------------------------------------------------------- /inc/buffer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/inc/buffer.h -------------------------------------------------------------------------------- /inc/calibrator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/inc/calibrator.h -------------------------------------------------------------------------------- /inc/helper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/inc/helper.h -------------------------------------------------------------------------------- /inc/model.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/inc/model.h -------------------------------------------------------------------------------- /inc/visualization.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/inc/visualization.h -------------------------------------------------------------------------------- /package.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/package.xml -------------------------------------------------------------------------------- /plugins/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/CMakeLists.txt -------------------------------------------------------------------------------- /plugins/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/README.md -------------------------------------------------------------------------------- /plugins/src/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/CMakeLists.txt -------------------------------------------------------------------------------- /plugins/src/FPSamplingPlugin/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/FPSamplingPlugin/CMakeLists.txt -------------------------------------------------------------------------------- /plugins/src/FPSamplingPlugin/FPSampling.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/FPSamplingPlugin/FPSampling.cu -------------------------------------------------------------------------------- /plugins/src/FPSamplingPlugin/FPSampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/FPSamplingPlugin/FPSampling.h -------------------------------------------------------------------------------- /plugins/src/FPSamplingPlugin/FPSamplingPlugin.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/FPSamplingPlugin/FPSamplingPlugin.cpp -------------------------------------------------------------------------------- /plugins/src/FPSamplingPlugin/FPSamplingPlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/FPSamplingPlugin/FPSamplingPlugin.h -------------------------------------------------------------------------------- /plugins/src/FPSamplingPlugin/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/FPSamplingPlugin/README.md -------------------------------------------------------------------------------- /plugins/src/ballQueryPlugin/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/ballQueryPlugin/CMakeLists.txt -------------------------------------------------------------------------------- /plugins/src/ballQueryPlugin/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/ballQueryPlugin/README.md -------------------------------------------------------------------------------- /plugins/src/ballQueryPlugin/ballQueryPlugin.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/ballQueryPlugin/ballQueryPlugin.cu -------------------------------------------------------------------------------- /plugins/src/ballQueryPlugin/ballQueryPlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/ballQueryPlugin/ballQueryPlugin.h -------------------------------------------------------------------------------- /plugins/src/common/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/common/CMakeLists.txt -------------------------------------------------------------------------------- /plugins/src/common/checkMacrosPlugin.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/common/checkMacrosPlugin.cpp -------------------------------------------------------------------------------- /plugins/src/common/checkMacrosPlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/common/checkMacrosPlugin.h -------------------------------------------------------------------------------- /plugins/src/common/common.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/common/common.cu -------------------------------------------------------------------------------- /plugins/src/common/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/common/common.h -------------------------------------------------------------------------------- /plugins/src/common/hash.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/common/hash.h -------------------------------------------------------------------------------- /plugins/src/common/plugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/common/plugin.h -------------------------------------------------------------------------------- /plugins/src/common/plugin_auto_declare.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/common/plugin_auto_declare.h -------------------------------------------------------------------------------- /plugins/src/common/print.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/common/print.h -------------------------------------------------------------------------------- /plugins/src/exports.map: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/exports.map -------------------------------------------------------------------------------- /plugins/src/gridBallQueryPlugin/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/gridBallQueryPlugin/CMakeLists.txt -------------------------------------------------------------------------------- /plugins/src/gridBallQueryPlugin/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/gridBallQueryPlugin/README.md -------------------------------------------------------------------------------- /plugins/src/gridBallQueryPlugin/gridBallQueryPlugin.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/gridBallQueryPlugin/gridBallQueryPlugin.cu -------------------------------------------------------------------------------- /plugins/src/gridBallQueryPlugin/gridBallQueryPlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/gridBallQueryPlugin/gridBallQueryPlugin.h -------------------------------------------------------------------------------- /plugins/src/havSamplingForQueryPlugin/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/havSamplingForQueryPlugin/CMakeLists.txt -------------------------------------------------------------------------------- /plugins/src/havSamplingForQueryPlugin/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/havSamplingForQueryPlugin/README.md -------------------------------------------------------------------------------- /plugins/src/havSamplingForQueryPlugin/havSamplingQ.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/havSamplingForQueryPlugin/havSamplingQ.cu -------------------------------------------------------------------------------- /plugins/src/havSamplingForQueryPlugin/havSamplingQ.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/havSamplingForQueryPlugin/havSamplingQ.h -------------------------------------------------------------------------------- /plugins/src/havSamplingForQueryPlugin/havSamplingQPlugin.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/havSamplingForQueryPlugin/havSamplingQPlugin.cu -------------------------------------------------------------------------------- /plugins/src/havSamplingForQueryPlugin/havSamplingQPlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/havSamplingForQueryPlugin/havSamplingQPlugin.h -------------------------------------------------------------------------------- /plugins/src/havSamplingPlugin/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/havSamplingPlugin/CMakeLists.txt -------------------------------------------------------------------------------- /plugins/src/havSamplingPlugin/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/havSamplingPlugin/README.md -------------------------------------------------------------------------------- /plugins/src/havSamplingPlugin/havSampling.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/havSamplingPlugin/havSampling.cu -------------------------------------------------------------------------------- /plugins/src/havSamplingPlugin/havSampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/havSamplingPlugin/havSampling.h -------------------------------------------------------------------------------- /plugins/src/havSamplingPlugin/havSamplingPlugin.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/havSamplingPlugin/havSamplingPlugin.cu -------------------------------------------------------------------------------- /plugins/src/havSamplingPlugin/havSamplingPlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/havSamplingPlugin/havSamplingPlugin.h -------------------------------------------------------------------------------- /plugins/src/nms3DPlugin/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/nms3DPlugin/CMakeLists.txt -------------------------------------------------------------------------------- /plugins/src/nms3DPlugin/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/nms3DPlugin/README.md -------------------------------------------------------------------------------- /plugins/src/nms3DPlugin/nms3D.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/nms3DPlugin/nms3D.cu -------------------------------------------------------------------------------- /plugins/src/nms3DPlugin/nms3D.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/nms3DPlugin/nms3D.h -------------------------------------------------------------------------------- /plugins/src/nms3DPlugin/nms3DPlugin.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/nms3DPlugin/nms3DPlugin.cpp -------------------------------------------------------------------------------- /plugins/src/nms3DPlugin/nms3DPlugin.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/plugins/src/nms3DPlugin/nms3DPlugin.h -------------------------------------------------------------------------------- /src/pcvt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/src/pcvt.py -------------------------------------------------------------------------------- /src/point_detector.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OuyangJunyuan/pointcloud-3d-detector-tensorrt/HEAD/src/point_detector.cpp --------------------------------------------------------------------------------