NTU Computer Vision Practice solution in Python