11
Aug
FORAY - Live Ball (with Distance) Detection
Aaron John Sabu foray computer vision machine learningI merged the ball detection code with the distance estimation code and added a camera component to make it live. The following video shows the code running live on CUDA:
Model prediction is very slow on the CPU; I am currently checking out techniques such as quantization to optimize the model. Also, although not depicted in the video, the ball detection algorithm detects a number of false positives; I will be adding more negative data to fix this.