LED Defect Detection (With Oriented Bounding Box)

Objective

This project, initiated by a company, aims to detect defects in LED installations on PCBs. LED defects are determined by the angle of installation; a non-defective LED is installed without tilting, while a tilted LED requires the AI model to generate an oriented bounding box, including its angle.

Due to the private nature of the dataset, no baseline results were available. Consequently, I established a new baseline and subsequently endeavored to enhance it by adapting the baseline model. The primary challenge lay in modifying the existing model to generate oriented bounding boxes, as the original model only produced normal bounding boxes with two annotation points.