ObjectLocalization

Code for detecting the pin for the case study

The code consist in the identification of a pin and transform its position to the robot frame so it can be gripped

There are three main file

• main.py for generating the artificial dataset
• CVPinTracking.py which uses openCV to identify the pin
• positionApproximation.py which generates a linear regression model for transforming the pixel coordinates to real world coordinates in the frame of the robot

Model to be integrated in the enviroment

The model is saved in the .h5 file and can be loaded using the positionApproxiamtionRun.py