Skip to content

Commit bf7a496

Browse files
lexavtankexmfcx
andauthored
docs(how-to-guides): add training docs for traffic_light_fine_detector (#489)
Signed-off-by: Aleksei Panferov <lexavtanke@gmail.com> Co-authored-by: Mete Fatih Cırıt <mfc@autoware.org>
1 parent 10fd0c7 commit bf7a496

File tree

1 file changed

+35
-0
lines changed

1 file changed

+35
-0
lines changed

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

Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,3 +40,38 @@ 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 a YOLOX model for `autoware_traffic_light_fine_detector`
45+
46+
To train a custom YOLOX model for use with the **Autoware Traffic Light Fine Detector**, please refer to the official YOLOX and Autoware training guides listed below. These documents provide the required setup, data preparation steps, and export instructions.
47+
48+
???+ abstract "Relevant documentation"
49+
50+
<div class="grid cards" markdown>
51+
52+
- **Autoware traffic_light_fine_detector README**
53+
54+
---
55+
56+
Training overview, model requirements, and ONNX export details:
57+
[autoware_universe/perception/autoware_traffic_light_fine_detector/README.md](https://github.com/autowarefoundation/autoware_universe/tree/main/perception/autoware_traffic_light_fine_detector/README.md)
58+
59+
- **YOLOX Custom Dataset Training Guide**
60+
61+
---
62+
63+
Instructions for preparing datasets, configuring experiments, and launching training:
64+
[YOLOX/train_custom_data.md](https://github.com/Megvii-BaseDetection/YOLOX/blob/main/docs/train_custom_data.md)
65+
66+
</div>
67+
68+
### 📦 Example Dataset (VOC Format)
69+
70+
To assist with training, an example dataset containing **1062 cropped traffic-light images with Pascal VOC annotations** is available here:
71+
72+
[:fa-cl-s fa-cloud-arrow-down: Download the sample traffic light dataset (3 MB)](https://autoware-files.s3.us-west-2.amazonaws.com/dataset/traffic_light_detection_sample_dataset.tar.gz){ .md-button }
73+
74+
- Use the YOLOX documentation to set up your environment and prepare your dataset (VOC or COCO).
75+
- Train a YOLOX model using your custom traffic-light data or the provided sample dataset.
76+
- After training, export the model to **ONNX** following the instructions in the Autoware `traffic_light_fine_detector` README.
77+
- Replace or integrate the exported ONNX model within the Autoware package for deployment.

0 commit comments

Comments
 (0)