Object Localisation using Machine Learning in Python