Color optimization and diffusion-based post-processing to obtain sharper images without compromising R-D performance in learned image compression