In one embodiment, a method includes obtaining an image comprising a plurality of pixels, determining, for a particular pixel of the plurality of pixels, a gradient value, classifying, based on the gradient value, the particular pixel into a flat class or one of a plurality of edge classes, and denoising the particular pixel based on the classification.