Skip to content

Commit 85e54b0

Browse files
authored
docs(how-to-guides): add training docs for centerpoint (#471)
Signed-off-by: Kaan Çolak <kaancolak95@gmail.com>
1 parent b772d90 commit 85e54b0

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

docs/how-to-guides/training-machine-learning-models/training-models.md

+10-3
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,15 @@ an example dataset containing three distinct classes (green, yellow, red), which
2828

2929
Detailed instructions for training the traffic light classifier model can be found **[here](https://github.com/autowarefoundation/autoware.universe/blob/main/perception/traffic_light_classifier/README.md)**.
3030

31-
<!--
31+
## Training CenterPoint 3D object detection model
3232

33-
Training traffic light detection model and lidar CenterPoint model will be added there.
33+
The CenterPoint 3D object detection model within the Autoware has been trained using the **[autowarefoundation/mmdetection3d](https://github.com/autowarefoundation/mmdetection3d/blob/main/projects/AutowareCenterPoint/README.md)** repository.
3434

35-
-->
35+
To train custom CenterPoint models and convert them into ONNX format for deployment in Autoware, please refer to the instructions provided in the README file included with Autoware's
36+
**[lidar_centerpoint](https://autowarefoundation.github.io/autoware.universe/main/perception/lidar_centerpoint/)** package. These instructions will provide a step-by-step guide for training the CenterPoint model.
37+
38+
In order to assist you with your training process, we have also included an example dataset in the TIER IV dataset format.
39+
40+
This dataset contains 600 lidar frames and covers 5 classes, including 6905 cars, 3951 pedestrians, 75 cyclists, 162 buses, and 326 trucks.
41+
42+
You can utilize this example dataset to facilitate your training efforts.

0 commit comments

Comments
 (0)