I 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.

Next Post Previous Post