From 62ce418c3c880bafa448defa2a4c151ad7f234a8 Mon Sep 17 00:00:00 2001 From: Aleksei Panferov Date: Tue, 28 Nov 2023 19:08:03 +0300 Subject: [PATCH] docs(how-to-guides): add training docs for traffic_light_fine_detector Signed-off-by: Aleksei Panferov --- .../training-models.md | 11 +++++++++++ 1 file changed, 11 insertions(+) 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..b46a26dc417 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 @@ -33,3 +33,14 @@ Detailed instructions for training the traffic light classifier model can be fou Training traffic light detection model and lidar CenterPoint model will be added there. --> + +## Training yolox detection model for Traffic_light_fine_detector package + +To train custom yolox traffic light detection models and convert them into ONNX format for deployment in Autoware, please refer to the instructions provided in the README file included with the +**"traffic_light_fine_detector"** package. These instructions will provide a step-by-step guide for training yolox model. + +In order to assist you with your training process, we have also included a sample dataset in the Pascal VOC format. +This dataset contains 1062 cropped images of traffic lights and annotations. +You can utilize **[this example dataset](https://autoware-files.s3.us-west-2.amazonaws.com/dataset/traffic_light_detection_sample_dataset.tar.gz)** to facilitate your training efforts. + +Detailed instructions for training the traffic light detector model can be found **[here](https://github.com/autowarefoundation/autoware.universe/blob/main/perception/traffic_light_fine_detector/README.md)**. \ No newline at end of file