Skip to content

Commit

Permalink
docs: update traffic light ssd README.md (autowarefoundation#2396)
Browse files Browse the repository at this point in the history
update traffic light ssd README.md

Signed-off-by: kminoda <koji.minoda@tier4.jp>
  • Loading branch information
Mingyu1991 authored and kminoda committed Jan 6, 2023
1 parent f269279 commit 4233fbc
Showing 1 changed file with 13 additions and 5 deletions.
18 changes: 13 additions & 5 deletions perception/traffic_light_ssd_fine_detector/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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

- <https://drive.google.com/uc?id=1USFDPRH9JrVdGoqt27qHjRgittwc0kcO>

## Inner-workings / Algorithms

Expand Down Expand Up @@ -48,10 +60,6 @@ Based on the camera image and the global ROI array detected by `map_based_detect

## Assumptions / Known limits

## Onnx model

- <https://drive.google.com/uc?id=1USFDPRH9JrVdGoqt27qHjRgittwc0kcO>

## Reference repositories

pytorch-ssd github repository
Expand Down

0 comments on commit 4233fbc

Please sign in to comment.