Automatic colorization of gray images is improved by optimizing color classes and balancing feature distribution.