diff --git a/perception/traffic_light_ssd_fine_detector/README.md b/perception/traffic_light_ssd_fine_detector/README.md index 87ff47c84085..3b283f35d245 100644 --- a/perception/traffic_light_ssd_fine_detector/README.md +++ b/perception/traffic_light_ssd_fine_detector/README.md @@ -4,7 +4,19 @@ It is a package for traffic light detection using MobileNetV2 and SSDLite. -The trained model is based on [pytorch-ssd](https://github.com/qfgaohao/pytorch-ssd). +## Training Information + +### Pretrained Model + +The model is based on [pytorch-ssd](https://github.com/qfgaohao/pytorch-ssd) and the pretrained model could be downloaded from [here](https://storage.googleapis.com/models-hao/mb2-ssd-lite-mp-0_686.pth). + +### Training Data + +The model was fine-tuned on 1750 TierIV internal images of Japanese traffic lights. + +### Trained Onnx model + +- ## Inner-workings / Algorithms @@ -48,10 +60,6 @@ Based on the camera image and the global ROI array detected by `map_based_detect ## Assumptions / Known limits -## Onnx model - -- - ## Reference repositories pytorch-ssd github repository