opencv中的copyTo函数在opencv-Python中的实现

小灰灰 2021-05-12 11:49 1141阅读 0赞

# Opencv中的copyTo()函数在opencv-Python中的实现 #

### copyTo() 函数有两种方式 ###

*  第一种：src\_Image.copyTo(dst\_Image)，表示将**原图**复制到**目标图**中。
 *  第二种：src\_Image.copyTo(dst\_Image, mask)，表示得到一个附加掩膜mask的**目标图**。

--------------------

第一种方法比较简单，这列主要讲解第二种方法。

### src\_Image.copyTo(dst\_Image, mask) ###

*  大致过程是说：将src\_Image图 对照着mask图复制到dst\_Image图，最后得到dst\_Image图。
 *  ##### 参数说明（原图需要和模板图有相同的大小，即高度，宽度，通道数都要相同，RGB图和GRAY图均可，只要他们相同大小，python中可用src\_Image.shape查看图像大小） #####
    
     *  src\_Image：原图
     *  dst\_Image：结果图
     *  mask：掩图

--------------------

copyTo()的实现原理:

*  在位置（x,y）时，如果mask的像素值不等于0，则dst(x,y) = src(x,y);  
    如果mask的像素值等于0，则dst(x,y) = 0
 *  即在(x,y)处，当mask的像素值是0时，dst对应的像素就为0；当mask的像素值不是0时，dst对应的像素值为scr的像素值（下图方便理解）  
    ![原理][20191017224402867.png]

--------------------

### 代码实现 ###

# 讲一下大致思路
    import cv2
    import numpy as np
    
    scr_Image = cv2.imread("原图")
    mask = cv2.imread("掩图")
    # 原图的高度和宽度
    height = scr_Image.shape[0]
    width = scr_Image.shape[1]
    # 生成和原图一样高度和宽度的矩形（全为0）
    dst_Image = np.zeros((height, width), np.uint8)
    
    # 以下是copyTo的算法原理：
    # 先遍历每行每列（如果不是灰度图还需遍历通道，可以事先把mask图转为灰度图）
    for row in range(mask.shape[0]):
        for col in range(mask.shape[1]):
            
            # 如果掩图的像素不等于0，则dst(x,y) = scr(x,y)
            if mask[row, col] != 0:
                 # dst_image和scr_Image一定要高宽通道数都相同，否则会报错
                dst_Image[row, col] = scr_Image[row, col]  
                
            # 如果掩图的像素等于0，则dst(x,y) = 0
            elif mask[row, col] == 0:
                dst_Image[row, col] = 0
    
    cv2.imshow("结果图", dst_Image)
    cv2.waitKey(0)
    cv2.destroyAllWindows()

#### 如果做的时候遇到如下的error，可能是图片的高宽通道数不一致，可以通过print(image.shape)来检查。 ####

—ValueError: setting an array element with a sequence.

[20191017224402867.png]: /images/20210511/6b63a15b2dd24ccead203d2ecbefa918.png