Skip to content

Commit

Permalink
fix: modify README.md
Browse files Browse the repository at this point in the history
Signed-off-by: liu cui <[email protected]>
  • Loading branch information
cyn-liu committed Nov 21, 2024
1 parent 4aec855 commit 1a5038b
Showing 1 changed file with 8 additions and 6 deletions.
14 changes: 8 additions & 6 deletions perception/autoware_tensorrt_bevdet/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -52,19 +52,21 @@ The core algorithm, named `BEVDet`, it unifies multi-view images into the perspe

Please refer to open source repository [ros2_dataset_bridge](https://github.com/Owen-Liuyuxuan/ros2_dataset_bridge) to publish the ROS 2 topics for NuScenes dataset.

## Limitation

The model is trained on open-source dataset `NuScenes` and has poor generalization on its own dataset, If you want to use this model to infer your data, you need to retrain it.

## Trained Models

You can download the onnx format of trained models by clicking on the links below.

- BEVDet: [bevdet_one_lt_d.onnx](https://drive.google.com/file/d/1eMGJfdCVlDPBphBTjMcnIh3wdW7Q7WZB/view?usp=sharing)

The model was trained in NuScenes database for 20 epochs.
The `BEVDet` model was trained in `NuScenes` dataset for 20 epochs.

## Limitation

The model is trained on open-source dataset `NuScenes` and has poor generalization on its own dataset, If you want to use this model to infer your data, you need to retrain it.

## Training BEVDet Model

If you want to train model using the [TIER IV's internal database(~2600 key frames)](https://drive.google.com/file/d/1UaarK88HZu09sf7Ix-bEVl9zGNGFwTVL/view?usp=sharing), please refer to the following repositories:[BEVDet adapted to TIER IV dataset](https://github.com/cyn-liu/BEVDet/tree/train_export)
If you want to train model using the [TIER IV's internal database(~2600 key frames)](https://drive.google.com/file/d/1UaarK88HZu09sf7Ix-bEVl9zGNGFwTVL/view?usp=sharing), please refer to the following repositories:[BEVDet adapted to TIER IV dataset](https://github.com/cyn-liu/BEVDet/tree/train_export).

## References/External links

Expand Down

0 comments on commit 1a5038b

Please sign in to comment.