Finetuning a pre-trained pytorch object detection model on traffic lights dataset.

• PyTorch 1.3 (Python 3.6)
• OpenCV

All *.py files expect tl_dataset.py are from pytorch/vision.

Traffic Light Detection sample data from Baidu Apollo Data Open Platform.

Please download the sample data and put it under ./apollo_tl_demo_data/.

See the train_tl_demo.ipynb file.

TorchVision Object Detection Finetuning Tutorial