diff --git a/docs/how-to-guides/.pages b/docs/how-to-guides/.pages index e4f75c009f9..c6582ed61fe 100644 --- a/docs/how-to-guides/.pages +++ b/docs/how-to-guides/.pages @@ -1,4 +1,5 @@ nav: - index.md - Integrating Autoware: integrating-autoware + - Training Machine Learning Models: training-machine-learning-models - others diff --git a/docs/how-to-guides/index.md b/docs/how-to-guides/index.md index dc05cd725cb..bc41dcd152a 100644 --- a/docs/how-to-guides/index.md +++ b/docs/how-to-guides/index.md @@ -4,6 +4,10 @@ - [Overview](integrating-autoware/overview.md) +## Training machine learning models + +- [Model Training and Deploying](training-machine-learning-models/training-models.md) + ## Others - [Debug Autoware](others/debug-autoware.md) diff --git a/docs/how-to-guides/training-machine-learning-models/.pages b/docs/how-to-guides/training-machine-learning-models/.pages new file mode 100644 index 00000000000..f5d7108726f --- /dev/null +++ b/docs/how-to-guides/training-machine-learning-models/.pages @@ -0,0 +1,2 @@ +nav: + - training-models.md 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 new file mode 100644 index 00000000000..eba497a1113 --- /dev/null +++ b/docs/how-to-guides/training-machine-learning-models/training-models.md @@ -0,0 +1,35 @@ +# Training Machine Learning Models + +## Overview + +The Autoware offers a comprehensive array of machine learning models, tailored for a wide range of tasks including 2D and 3D object detection, +traffic light recognition and more. These models have been meticulously trained utilizing **[open-mmlab](https://github.com/open-mmlab)**'s extensive repositories. +By leveraging the provided scripts and following the training steps, you have the capability to train these models using your own dataset, +tailoring them to your specific needs. + +Furthermore, you will find the essential conversion scripts to deploy your trained models into Autoware using the **[mmdeploy](https://github.com/open-mmlab/mmdeploy)** repository. + +## Training traffic light classifier model + +The traffic light classifier model within the Autoware has been trained using the **[mmlab/pretrained](https://github.com/open-mmlab/mmpretrain)** repository. +The Autoware offers pretrained models based on EfficientNet-b1 and MobileNet-v2 architectures. +To fine-tune these models, a total of 83,400 images were employed, comprising 58,600 for training, +14,800 for evaluation, and 10,000 for testing. These images represent Japanese traffic lights and were trained using TIER IV's internal dataset. + +| Name | Input Size | Test Accuracy | +| --------------- | ---------- | ------------- | +| EfficientNet-b1 | 128 x 128 | 99.76% | +| MobileNet-v2 | 224 x 224 | 99.81% | + +Comprehensive training instructions for the traffic light classifier model are detailed within +the readme file accompanying **"traffic_light_classifier"** package. These instructions will guide you through +the process of training the model using your own dataset. To facilitate your training, we have also provided +an example dataset containing three distinct classes (green, yellow, red), which you can leverage during the training process. + +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)**. + +