前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >OpenCV-Python实战(1) —— 给图片添加图片水印【利用 OpenCV 像素的读写原理实现】

OpenCV-Python实战(1) —— 给图片添加图片水印【利用 OpenCV 像素的读写原理实现】

作者头像
Rattenking
发布2022-10-24 16:04:53
1.5K0
发布2022-10-24 16:04:53
举报
文章被收录于专栏:Rattenking

1. 需求分析

  1. 使用 OpenCV 图像像素读写操作原理,也就是 imagerow 获取和设值;
  2. 给图片添加水印:对应添加水印的位置,设置为水印图片的像素;
  3. 水印下能看到原图,也就是说水印必须是透明图片;
  4. 同时将水印透明位置像素(0,0,0)改变为同位置的图片的像素。

2. 直接将水印添加到图片

2.1 代码逻辑分析
  1. 读取水印logo图片;
  2. 读取需要添加水印的图片;
  3. 获取水印的宽高; 注意:此处水印使用的三通道读取,因此获取水印属性时需要加通道值。
  4. 利用图像像素读取设值原理,直接修改对应像素为水印像素;
  5. 注意:此处默认直接从坐标(0,0)开始直接修改【需要添加水印的图片】,如果其他位置,需要获取【需要添加水印的图片】的宽高等属性
  6. 显示添加水印后的图像。
2.2 代码实现
代码语言:javascript
复制
import cv2 as cv

def add_logo_to_img():
  # 水印logo图片
  logo = cv.imread('./images/opencv-logo-white.png')
  # 需要添加水印的图片
  img = cv.imread('./images/messi5.jpg')
  # 获取水印的宽高,注意此处水印使用的三通道读取,因此获取水印属性时需要加通道值
  h,w,c = logo.shape
  # 利用图像像素读取设值原理,直接修改对应像素为水印像素
  img[:h,:w] = logo
  # 显示添加水印后的图像
  cv.imshow('add_logo', img)
  cv.waitKey(0)
  cv.destroyAllWindows()

if __name__ == "__main__":
  add_logo_to_img()
2.3 实现结果
2.4 注意
  1. 此处水印使用的三通道读取,因此获取水印属性时需要加通道值。
  2. 注意:此处默认直接从坐标(0,0)开始直接修改。
  3. 此处的水印图片是透明的水印图片,因此加上水印后会看到水印周边很多黑色。
  4. 水印图片的宽高比原图尺寸小。
2.5 非透明水印添加代码
代码语言:javascript
复制
import cv2 as cv

def add_logo_to_img():
  # 水印logo图片
  logo = cv.imread('./images/butterfly-mall.jpg')
  # 需要添加水印的图片
  img = cv.imread('./images/messi5.jpg')
  # 获取水印的宽高,注意此处水印使用的三通道读取,因此获取水印属性时需要加通道值
  h,w,c = logo.shape
  # 利用图像像素读取设值原理,直接修改对应像素为水印像素
  img[:h,:w] = logo
  # 显示添加水印后的图像
  cv.imshow('add_logo', img)
  cv.waitKey(0)
  cv.destroyAllWindows()

if __name__ == "__main__":
  add_logo_to_img()
2.5 非透明水印添加结果

3. 遍历像素添加水印

3.1 代码逻辑分析
  1. 读取水印logo图片;
  2. 读取需要添加水印的图片;
  3. 获取水印的宽高; 注意:此处水印使用的三通道读取,因此获取水印属性时需要加通道值。
  4. 利用图像像素读取设值原理,直接修改对应像素为水印像素;
  5. 遍历 logo 的像素,发现像素值是黑色(0,0,0)就替换为对应的原图像素【反之就是logo像素不是黑色的位置直接赋值给原图】;
  6. 注意:此处默认直接从坐标(0,0)开始直接修改【需要添加水印的图片】,如果其他位置,需要获取【需要添加水印的图片】的宽高等属性
  7. 显示添加水印后的图像。
3.2 代码实现
代码语言:javascript
复制
def add_logo_to_img():
  # 水印logo图片
  logo = cv.imread('./images/opencv-logo-white.png')
  # 需要添加水印的图片
  img = cv.imread('./images/messi5.jpg')
  # 获取水印的宽高,注意此处水印使用的三通道读取,因此获取水印属性时需要加通道值
  h,w,c = logo.shape
  # 利用图像像素读取设值原理,直接修改对应像素为水印像素
  # 遍历 logo 的像素,发现像素值是黑色(0,0,0)就替换为对应的原图像素【反之就是logo像素不是黑色的位置直接赋值给原图】
  for row in range(h):
    for col in range(w):
      b,g,r = logo[row][col]
      if b != 0 or g != 0 or r != 0:
        img[row][col] = (b,g,r)
  
  # 显示添加水印后的图像
  cv.imshow('add_logo', img)
  cv.waitKey(0)
  cv.destroyAllWindows()

if __name__ == "__main__":
  add_logo_to_img()
3.3 实现结果

4. 改变水印的位置【居中】

4.1 代码逻辑分析
  1. 在原来的基础上【获取原图的宽高属性】;
  2. 计算水印居中时的其实坐标;
  3. 修改像素值时,加上起始坐标的值,才是改变对应原图的像素坐标。
4.2 代码实现
代码语言:javascript
复制
import cv2 as cv

def add_logo_to_img():
  # 水印logo图片
  logo = cv.imread('./images/opencv-logo-white.png')
  # 需要添加水印的图片
  img = cv.imread('./images/messi5.jpg')
  # 获取水印的宽高,注意此处水印使用的三通道读取,因此获取水印属性时需要加通道值
  h,w,c = logo.shape
  # 获取原图的宽高属性
  mh,mw,mc = img.shape
  # 计算水印居中时的其实坐标
  y = int((mh - h) / 2)
  x = int((mw - w) / 2)
  # 利用图像像素读取设值原理,直接修改对应像素为水印像素
  # 遍历 logo 的像素,发现像素值是黑色(0,0,0)就替换为对应的原图像素【反之就是logo像素不是黑色的位置直接赋值给原图】
  for row in range(h):
    for col in range(w):
      b,g,r = logo[row][col]
      if b != 0 or g != 0 or r != 0:
        img[row + y][col + x] = (b,g,r)
  
  # 显示添加水印后的图像
  cv.imshow('add_logo', img)
  cv.waitKey(0)
  cv.destroyAllWindows()

if __name__ == "__main__":
  add_logo_to_img()
4.3 实现结果

5. 使用整体法实现水印的位置的改变

5.1 代码逻辑分析
  1. 在第四步的基础上添加【截取添加水印位置和水印等大的图像】;
  2. 此时截取图像和水印图像大小一致;
  3. 直接使用第二步的方法,直接修改这个大小一致的截取图像,此处的方便在于不用遍历的时候每次都需要计算修改的位置
  4. 将修改后的截取图像还原回原图像;
  5. 实现添加水印效果。
5.2 实现代码
代码语言:javascript
复制
import cv2 as cv

def add_logo_to_img():
  # 水印logo图片
  logo = cv.imread('./images/opencv-logo-white.png')
  # 需要添加水印的图片
  img = cv.imread('./images/messi5.jpg')
  # 获取水印的宽高,注意此处水印使用的三通道读取,因此获取水印属性时需要加通道值
  h,w,c = logo.shape
  # 获取原图的宽高属性
  mh,mw,mc = img.shape
  # 计算水印居中时的其实坐标
  y = int((mh - h) / 2)
  x = int((mw - w) / 2)
  # 截取和水印大小等同的原图需要改变位置的图像
  change_img = img[y:y+h,x:x+w]
  # 显示添加水印位置的截取图像
  cv.imshow('change_img', change_img)
  # 修改截取图片
  # 利用图像像素读取设值原理,直接修改对应像素为水印像素
  # 遍历 logo 的像素,发现像素值是黑色(0,0,0)就替换为对应的原图像素【反之就是logo像素不是黑色的位置直接赋值给原图】
  for row in range(h):
    for col in range(w):
      b,g,r = logo[row][col]
      if b != 0 or g != 0 or r != 0:
        change_img[row][col] = (b,g,r)
  # 显示截取图像添加水印后的效果
  cv.imshow('change_img_add_logo', change_img)
  # 将截取的图片还原到截取位置
  img[y:y+h,x:x+w] = change_img
  # 显示添加水印后的图像
  cv.imshow('add_logo', img)
  cv.waitKey(0)
  cv.destroyAllWindows()

if __name__ == "__main__":
  add_logo_to_img()
5.3 实现结果

6. 总结

  1. 本篇文章是学习【图像像素读写操作】的实战实现;
  2. 文章还差水印文字的实现,水印图片大小的改变实现,在后续文章中学到了依旧会使用到开发实例中;
  3. 注意本文章都没有考虑图片的超出限制,如果是需要logo的部分等操作,需要另行操作;
  4. 注意本文章没有使用cv的其他二值处理等函数处理水印,所以水印最好是透明背景的图片,或者是纯色背景,方便获取背景的像素值。
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2022-10-21,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 需求分析
  • 2. 直接将水印添加到图片
    • 2.1 代码逻辑分析
      • 2.2 代码实现
        • 2.3 实现结果
          • 2.4 注意
            • 2.5 非透明水印添加代码
              • 2.5 非透明水印添加结果
              • 3. 遍历像素添加水印
                • 3.1 代码逻辑分析
                  • 3.2 代码实现
                    • 3.3 实现结果
                    • 4. 改变水印的位置【居中】
                      • 4.1 代码逻辑分析
                        • 4.2 代码实现
                          • 4.3 实现结果
                          • 5. 使用整体法实现水印的位置的改变
                            • 5.1 代码逻辑分析
                              • 5.2 实现代码
                                • 5.3 实现结果
                                • 6. 总结
                                领券
                                问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档