An improved automatic white balance algorithm based on the feature of Bayer images which contain less information than color images, is proposed in this paper. By calculating the corrected parameters of auto white balance, this algorithm reduces the computation time by half of that of Gray World-Retinex, and obtains ideal color images quickly. The movement details of oil droplets in the lithography model can be observed by applying this method to the micro-water/oil displacement system.