Skip to content

Commit

Permalink
docs(how-to-guides): add training docs for traffic_light_fine_detector
Browse files Browse the repository at this point in the history
Signed-off-by: Aleksei Panferov <[email protected]>
  • Loading branch information
lexavtanke committed Nov 29, 2023
1 parent f508499 commit 62ce418
Showing 1 changed file with 11 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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)**.

0 comments on commit 62ce418

Please sign in to comment.