GEMstack [s2024_road] Adding code that trains the model used traffic sign and traffic light detection module

[s2024_road] Adding code that trains the model used traffic sign and traffic light detection module

Open ananaymathur opened this issue 9 months ago • 2 comments

This PR adds:

traffic_sign_and_light_model_train.py to GEMstack/offboard/detection_learning/
This is the code used to train the custom object detection model used in the traffic sign and traffic light detection module
Code also runs the validation set

May 06 '24 22:05 ananaymathur

data.yaml is part of the dataset that the model is being trained on. I did not push the dataset to the repo. I can add a readme explaining what the code does and also link the dataset in there.

May 07 '24 06:05 ananaymathur

You should indeed add that documentation if your code is to be approved.