[TASK] Train yolo model for docking localization

[philliab13]

Description of task

Train a yolo model to detect the dock. This is not a hard task; it requires some research and labeling of data.

Here you have to set the drone in different orientations to create our own training data, then you will have to label which points are the dock and not, manually.

Suggested Workflow

1. Find out how to set up a yolo model
2. Gather data
3. Label data
4. Train model

Specifications

1. Working yolo model that detects the dock at different positions

Contacts

No response

Code Quality

• Every function in header files are documented (inputs/returns/exceptions)
• The project has automated tests that cover MOST of the functions and branches in functions (pytest/gtest)
• The code is documented on the wiki (provide link)