A colour transfer method of interior design based on machine learning

Abstract

In order to overcome the problems of large colour range, low structure similarity and poor objective evaluation index in the process of interior design colour transfer, this paper proposes a new method of interior design colour transfer based on machine learning. In this method, K-means algorithm is introduced to eliminate the uneven brightness area of the target image. The target image is processed by initial clustering, and the iterative threshold segmentation method is used to obtain the final clustering accurate target image. Combined with machine learning, the corresponding samples are selected on the two images to finish colouring, and the colour transfer of interior design is realised by referring to the coloured sample block. The experimental results show that the colour degree of the proposed method is maintained between 33 and 45, the structural similarity is always above 95%, and the comprehensive objective evaluation index value is close to 100%.