Skip to content

Commit b5579b7

Browse files
lexavtankexmfcx
authored andcommitted
docs(how-to-guides): add training docs for traffic_light_fine_detector
Signed-off-by: Aleksei Panferov <lexavtanke@gmail.com>
1 parent 10fd0c7 commit b5579b7

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

docs/tutorials/training-machine-learning-models/training-models.md

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,3 +40,14 @@ In order to assist you with your training process, we have also included an exam
4040
This dataset contains 600 lidar frames and covers 5 classes, including 6905 cars, 3951 pedestrians, 75 cyclists, 162 buses, and 326 trucks.
4141

4242
You can utilize this example dataset to facilitate your training efforts.
43+
44+
## Training yolox detection model for Traffic_light_fine_detector package
45+
46+
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
47+
**"traffic_light_fine_detector"** package. These instructions will provide a step-by-step guide for training yolox model.
48+
49+
In order to assist you with your training process, we have also included a sample dataset in the Pascal VOC format.
50+
This dataset contains 1062 cropped images of traffic lights and annotations.
51+
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.
52+
53+
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)**.

0 commit comments

Comments
 (0)