Update smoothen_orientation_image.py