Could this code work to implement keypoint regression on gates for drone racing? Of course, providing a proper training set.