加水印應(yīng)該是個(gè)很常見(jiàn)的需求，但是網(wǎng)上找的代碼，都感覺(jué)不太完善。記錄下自己搞出來(lái)的一個(gè)方案

水印有幾個(gè)需求：

1. 中文文字水印
2. 文字傾斜
3. 滿圖都是，而不是只有一個(gè)地方
4. 水印文字所在之處完全展示水印

實(shí)現(xiàn)思路

準(zhǔn)備水印圖

我是這么做的，先手工生成一張水印圖，尺寸比較小，約100*100。然后也把文字旋轉(zhuǎn)一定角度（當(dāng)然如果想做隨機(jī)角度，也是可以的，代碼再?gòu)?fù)雜點(diǎn)處理就好，不影響這個(gè)思路），生成這個(gè)一張圖片，并且底色選擇純黑。

然后用opencv將水印圖處理出一張二值圖

這個(gè)也不難，底色純黑，這個(gè)很好做，用Threshold很容易實(shí)現(xiàn)。

然后用水印圖和原圖疊加

核心就是要用copyTo()，然后要輸入mask，二值圖的作用就是做掩碼，這樣就可以實(shí)現(xiàn)完美的水印添加效果。

其他細(xì)節(jié)

因?yàn)樗D通常是遠(yuǎn)小于原圖的（其實(shí)也可以采用其他方案。總之核心就是要有水印圖和對(duì)應(yīng)的二值圖），如何實(shí)現(xiàn)滿圖添加水印呢？

思路

1. 先設(shè)置一個(gè)水印的間距，比如兩百個(gè)像素點(diǎn)
2. 然后通過(guò)水印圖的大小和間距的值，計(jì)算生成一個(gè)尺寸大于等于原圖的純水印圖（同時(shí)也要有二值圖）
3. 然后裁剪成和原圖一樣大
4. 然后用copyTo()疊加即可。

這樣性能很高，go語(yǔ)言實(shí)現(xiàn)處理一張圖片不超過(guò)10ms。

go語(yǔ)言實(shí)現(xiàn)

思路都是通用的，基于opencv都能實(shí)現(xiàn)。

func ImgWatermark(img gocv.Mat) {var Watermark gocv.Matvar WatermarkBW gocv.Mat// 不同通道的圖片使用不同的水印if img.Channels() == 3 {// 判斷是否需要初始化if g.Watermark3 == nil { // g. 是我設(shè)置的全局變量，因?yàn)椴恍枰看味技虞d水印。加載一次到內(nèi)存即可mat := gocv.IMRead(`watermark.jpg`, gocv.IMReadColor)g.Watermark3 = &mat// 取得水印圖片的二值圖watermarkGray3 := gocv.NewMat()gocv.CvtColor(*g.Watermark3, &watermarkGray3, gocv.ColorBGRToGray)// 取得二值圖的黑白圖BW3 := gocv.NewMat()g.WatermarkBW3 = &BW3gocv.Threshold(watermarkGray3, g.WatermarkBW3, 30, 255, gocv.ThresholdBinary)watermarkGray3.Close()}Watermark = *g.Watermark3WatermarkBW = *g.WatermarkBW3} else if img.Channels() == 4 {// 判斷是否需要初始化if g.Watermark4 == nil {mat := gocv.IMRead(`watermark.jpg`, gocv.IMReadColor)g.Watermark4 = &matgocv.Merge([]gocv.Mat{mat, gocv.NewMatWithSizeFromScalar(gocv.NewScalar(255, 255, 255, 255), mat.Rows(), mat.Cols(), gocv.MatTypeCV8UC1)}, g.Watermark4)// 取得水印圖片的二值圖watermarkGray4 := gocv.NewMat()gocv.CvtColor(*g.Watermark4, &watermarkGray4, gocv.ColorBGRToGray)// 取得二值圖的黑白圖BW4 := gocv.NewMat()g.WatermarkBW4 = &BW4gocv.Threshold(watermarkGray4, g.WatermarkBW4, 30, 255, gocv.ThresholdBinary)watermarkGray4.Close()}Watermark = *g.Watermark4WatermarkBW = *g.WatermarkBW4} else if img.Channels() == 1 {// 判斷是否需要初始化if g.Watermark1 == nil {mat := gocv.IMRead(`watermark.jpg`, gocv.IMReadGrayScale) // 直接讀取黑白g.Watermark1 = &mat// 取得二值圖的黑白圖BW1 := gocv.NewMat()g.WatermarkBW1 = &BW1gocv.Threshold(mat, g.WatermarkBW1, 30, 255, gocv.ThresholdBinary)}Watermark = *g.Watermark1WatermarkBW = *g.WatermarkBW1} else {fmt.Println("圖片通道數(shù)不支持", img.Channels())return}waterCol := Watermark.Cols()waterRow := Watermark.Rows()// 為了水印不要那么密，這里設(shè)置一個(gè)間距spacing := 200 // 水印的間距waterColAndSpacing := waterCol + spacingwatreRowAndSpacing := waterRow + spacing// 計(jì)算需要添加水印的次數(shù)watermarkHugeRowTimes := int(math.Ceil(float64(img.Rows()) / float64(watreRowAndSpacing))) //向上取整watermarkHugeColTimes := int(math.Ceil(float64(img.Cols()) / float64(waterColAndSpacing)))imgRect := image.Rectangle{} //每次取圖像中哪一塊區(qū)域的數(shù)據(jù)結(jié)構(gòu)var croppedMat gocv.Mat      // 每次從原圖像上裁剪和水印圖像一樣大小的一塊newWater := WatermarknewWaterBW := WatermarkBW// 循環(huán)一塊塊的給圖像添加水印for i := 0; i < watermarkHugeColTimes; i++ {imgRect.Min.X = i * waterColAndSpacingimgRect.Max.X = imgRect.Min.X + waterColif imgRect.Max.X >= img.Cols() { // 判斷是否超出原圖像的列數(shù)imgRect.Max.X = img.Cols()}for j := 0; j < watermarkHugeRowTimes; j++ {imgRect.Min.Y = j * watreRowAndSpacingimgRect.Max.Y = imgRect.Min.Y + waterRowif imgRect.Max.Y >= img.Rows() { // 判斷是否超出原圖像的行數(shù)imgRect.Max.Y = img.Rows()}croppedMat = img.Region(imgRect) // 原圖像中一塊區(qū)域的淺拷貝，修改它會(huì)連帶修改原圖像// 判斷裁剪的圖像大小是否與水印圖像大小一致，不一致則需要重新裁剪if croppedMat.Rows() != Watermark.Rows() || croppedMat.Cols() != Watermark.Cols() {newRect := image.Rectangle{Min: image.Point{X: 0, Y: 0}, Max: image.Point{X: croppedMat.Cols(), Y: croppedMat.Rows()}}newWater = Watermark.Region(newRect)newWaterBW = WatermarkBW.Region(newRect)}newWater.CopyToWithMask(&croppedMat, newWaterBW) // 用水印圖覆蓋原圖像}}
}