Improving image quality with a neural network