diff --git a/docs/how-to-guides/training-machine-learning-models/training-models.md b/docs/how-to-guides/training-machine-learning-models/training-models.md index fa692c53666..4f9e55cfef5 100644 --- a/docs/how-to-guides/training-machine-learning-models/training-models.md +++ b/docs/how-to-guides/training-machine-learning-models/training-models.md @@ -28,8 +28,15 @@ an example dataset containing three distinct classes (green, yellow, red), which 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)**. - +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 +**[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. + +In order to assist you with your training process, we have also included an example dataset in the TIER IV dataset format. + +This dataset contains 600 lidar frames and covers 5 classes, including 6905 cars, 3951 pedestrians, 75 cyclists, 162 buses, and 326 trucks. + +You can utilize this example dataset to facilitate your training efforts.