Object Location Prediction

Predict the (x, y, z) coordinates of an object in a 3D space using Neural Nets.

This is a video showing the working of the model in real time. We piped the predictions(x, y, z) made by the Neural Net into a Holo Lens to generate a virtual object(red sphere) at the position predicted by the NN.

Dependencies

• keras
• tensorflow

Dataset(based on python version)

Download any one of the following and place it in the "data" directory.

• For Python 2 (download both pickles)
• For Python 3

More information

You can find more information about our CNN here.