The color of "Huashan Rock Painting" can be integrated with cultural and creative design under AI drive, further enriching the style and connotation of cultural and creative design. This article extracts the color elements contained in "Huashan Rock Paintings" through the Stylegan model and high-resolution collection, and provides more possibilities for the application of traditional art in contemporary cultural and creative industries through case studies. Research has shown that design forms represented by AI not only enhance the efficiency of reshaping traditional art, but also achieve a balance of social, cultural, and economic values to a certain extent, providing a new paradigm for the inheritance and dissemination of rock art cultural heritage for reference.