OpenCV | OpenCV 圖檔換底色

環境

pip install opencv-python==3.4.2.16
pip install opencv-contrib-python==3.4.2.16      

示例代碼

import cv2
import  numpy as np
 
img=cv2.imread('molecule.png')
#縮放
rows,cols,channels = img.shape
img=cv2.resize(img,None,fx=0.5,fy=0.5)
rows,cols,channels = img.shape
cv2.imshow('img',img)
 
#轉換hsv
hsv=cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
lower_blue=np.array([90,70,70])
upper_blue=np.array([110,255,255])
mask = cv2.inRange(hsv, lower_blue, upper_blue)
cv2.imshow('Mask', mask)
 
#腐蝕膨脹
erode=cv2.erode(mask,None,iterations=1)
cv2.imshow('erode',erode)
dilate=cv2.dilate(erode,None,iterations=1)
cv2.imshow('dilate',dilate)
 
#周遊替換
for i in range(rows):
    for j in range(cols):
        if dilate[i,j]==255:
            img[i,j]=(0,0,255)#此處替換顔色，為BGR通道
cv2.imshow('res',img)
 
cv2.waitKey(0)
cv2.destroyAllWindows()      

原圖

輸出圖