Python find shape in image