In [ ]:
# Undistrotion

output = cv2.undistort(image, mtx, dist, None, mtx)
#cv2.imwrite('undistorted.jpg',output)
cv2.imshow("image",image)
cv2.imshow("output",output)
cv2.waitKey(0);
cv2.destroyAllWindows()