/BEVDet

A fork of the BEVDet series .

Primary LanguagePythonApache License 2.0Apache-2.0

A fork of BEVDet

English | 简体中文

This repository provides the script for exporting onnx required by bevdet-tensorrt-cpp.

Environment

Please refer to BEVDet and mmdetection3d.

Running

export onnx

python tools/export/export_onnx.py configs/bevdet/bevdet-r50-4dlongterm-depth-cbgs.py <checkpoint> --postfix='_lt_d' 

This repository provides checkpoint.

export yaml required by bevdet-tensorrt-cpp

python tools/export/export_yaml.py configs/bevdet/bevdet-r50-4dlongterm-depth-cbgs.py --prefix='bevdet_lt_d'

test

python tools/export/test.py configs/bevdet/bevdet-r50-4dlongterm-depth-cbgs.py path_to_results