Learning 2D geometric shapes