Are the traffic light and tensorrt_yolo training datasets available?

[tleyden]

I'm looking for ways to contribute to the autoware perception stack by improving the accuracy of the models. Are the training datasets available for any of these modules?

• traffic_light_classifier
• traffic_light_ssd_fine_detector
• tensorrt_yolo

I did some ad hoc testing against the traffic_light_classifier model with GPU/CNN enabled. It worked fairly well on a small batch of real-world image data that I collected using my mobile device, but it gave incorrect predictions on some images I downloaded from google images. I could provide the test samples if they'd be useful.

Was tensorrt_yolo trained on the coco dataset or a custom dataset?