Illumination correction of dyed fabrics approach using Bagging-based ensemble particle swarm optimization–extreme learning machine

Abstract

Changes in illumination will result in serious color difference evaluation errors during the dyeing process. A Bagging-based ensemble extreme learning machine (ELM) mechanism hybridized with particle swarm optimization (PSO), namely Bagging–PSO–ELM, is proposed to develop an accurate illumination correction model for dyed fabrics. The model adopts PSO algorithm to optimize the input weights and hidden biases for the ELM neural network called PSO–ELM, which enhances the performance of ELM. Meanwhile, to further increase the prediction accuracy, a Bagging ensemble scheme is used to construct an independent PSO–ELM learning machine by taking bootstrap replicates of the training set. Then, the obtained multiple different PSO–ELM learners are aggregated to establish the prediction model. The proposed prediction model is evaluated with real dyed fabric images and discussed in comparison with several related methods. Experimental results show that the ensemble color constancy method is able to generate a more robust illuminant estimation model with better generalization performance.