python image restoration