pyqt5与opencv对照片换底色

前言

库

opencv-python

pyqt5

换色原理

直言的说，在这篇大佬的文章中，

(32条消息) Python 教你用OpenCV实现给照片换底色_叶庭云的博客-CSDN博客

可以实现换底色，我认为其中最关键的一步

binary_img = cv2.inRange(gray_img, low_value, high_value)
# 这个函数inRange

这个函数--相当于是二值化函数，在low_value与high_value之间的变成255，否则变成0

效果还是很好的，但是就只有蓝色换成其他颜色，比如，我想把白色底色的图片换成其他颜色怎么办？？

pyqt5处理边缘

直言的说。我参考了这篇大佬的博客

(32条消息) Python12 PyQt5实现鼠标移动绘制图（自定义画笔粗细和颜色，橡皮擦功能，保存文件）_pyqt5 鼠标移动后固定颜色_今年的浙江省状元的博客-CSDN博客

这篇博客，我觉得---最关键的一步

class PaintBoard(QWidget):
    ...

创建一个画板（新的窗口），与主窗口进行交互，主窗口控制颜色，大小之类的参数。传到画板中。

简单的说：不同窗口的交互。

这就很重要了。

正文

换色的页面

import cv2 as cv
import numpy as np
class change_color:
    def __init__(self,picture):
        """
        :param picture: 已经被imread读取的图片
        """
        self.iter = None
        self.picture=picture
        self.color=None
        self.num=None
    def choice_mode(self,num):
# 选择模式，对底色不同的图片，实际上模式是不一样的
        if num==1:
            return cv.MORPH_CLOSE
        else:
            return cv.MORPH_OPEN
    def set_num(self,num):
# 设置迭代次数
        self.num=num
    def get_color(self,color):
# 得到颜色
        self.color=color
    def white(self):
# 白色的范围
# 这个感觉要自己调节，自己尝试
        min_value = np.array([0,5, 5])
        max_value = np.array([255, 255, 255])
        return [min_value,max_value]
    def blue(self):
# 这个范围就用第一位大佬的博客的范围
        min_value = np.array([90, 70, 70])
        max_value = np.array([110, 255, 255])
        return [min_value,max_value]
    def start(self,color,what):
# what 确定原来的底色，这一步参考了大佬的博客
        rows, cols, channels = self.picture.shape
        gray_img = cv.cvtColor(self.picture, cv.COLOR_BGR2HSV)
        binary_img = cv.inRange(gray_img, color()[0], color()[1])
        if what=='白色':
# 如果底色是白色的，需要对二值化后的照片进行反色处理
            binary_img=255-binary_img
        kernel=cv.getStructuringElement(cv.MORPH_RECT,(5,5))
        mv1=cv.morphologyEx(binary_img,self.choice_mode(self.num),kernel,iterations=self.iter)
# 分别确定是开运算，还是闭运算，其实也可以对卷积核进行设置。
        for i in range(rows):
            for j in range(cols):
                if mv1[i, j] == 255:
                    self.picture[i, j] = self.color
# 对颜色的替换，参考了第一位大佬的博客，但是颜色可以任意，由pyqt5来选择
    def return_picture(self):
# 把处理后的图片进行其他处理，边缘处理
        self.picture=cv.cvtColor(self.picture,cv.COLOR_BGR2RGB)
        return self.picture
    def set_iter(self,iter):
# 设置迭代次数
        self.iter=iter
    def main(self,what):
# 对不同底色的图片进行选择
        match what:
            case '白色':
                self.start(self.white,what)
            case '蓝色':
                self.start(self.blue,what)

实际上底色还可以由红色，这里就没写出来，但应该差不多

对白色的底色变成其他颜色的底色，需要进行一步 反色操作，（自己尝试过就会知道为什么要反色）

😛😛😛

我用了最简单的方法

binary_img=255-binary_img

肯定有其他方法。可以去搜一搜，这里就不说明了。

pyqt5的页面

import os
import cv2 as cv
import numpy as np
import sys
from PyQt5.QtWidgets import *
from PyQt5.QtGui import *
from PyQt5.QtCore import *
from color_base import change_color

class cv_Board(QWidget):
# 创建画板，传入参数
    def __init__(self,picture,spin,color=Qt.black):
        super().__init__()
        self.paint = None
        self.spin=spin
        self.color=color
        self.picture=picture
        self.initui()
    def initui(self):
        self.paint=QPainter(self)
    def paintEvent(self, e):
        if self.picture:
            self.paint.begin(self)
            self.paint.drawPixmap(0,0,self.picture)
            self.paint.end()
    def get_picture(self,picture):
        self.clear()
        self.picture=picture
    def clear(self):
        self.picture.fill(Qt.white)
    def mouseMoveEvent(self, e):
        if self.picture:
            self.paint.begin(self.picture)
            self.paint.setPen(QPen(self.color, self.spin))
            self.paint.drawPoint(e.pos())
            self.paint.end()
            self.update()
    def change_spin(self,spin):
        self.spin=spin
    def get_color(self,color):
        self.color=color
    def save(self):
        image = self.picture.toImage()
        return image

class Main(QWidget):
# 核心
    def __init__(self,picture):
        super().__init__()
        self.nums = None
        self.iters = None
        self.picture = None
        self.cv_img = None
        self.color = None
        self.color_change = None
        self.board = None
        self.size = None
        self.spin = None
        self.pen_window=None
        self.paint = None
        self.img = None
        self.filename = picture
        self.initui()
    def initui(self):
        self.setWindowTitle('核心')
        self.set_layout()
    def set_layout(self):
        h=QHBoxLayout(self)
        h1=QHBoxLayout(self)
        v2=QVBoxLayout(self)
        g=QGridLayout(self)
        p=self.get_widget()
        m=n=0
        for i in p:
            g.addWidget(i,m,n,alignment=Qt.AlignTop)
            if m>=n:
                n+=1
            else:
                m,n=n,m
        a=QSpacerItem(20,20)
        g.addLayout(v2,3,0)
        w1=self.get_widget2()
        for i in w1:
            v2.addWidget(i)
        h1.addWidget(self.get_paint())
        h.addLayout(h1)
        h.addLayout(g)
        return h
    def get_widget(self):
        label1=QLabel('笔的大小',self)
        self.spin=QSpinBox(self)
        self.spin.setValue(5)
        self.size=5
        self.spin.valueChanged.connect(self.spin_value)
        label2=QLabel('颜色',self)
        self.color_change=QPushButton('选择颜色',self)
        self.color_change.clicked.connect(self.color_value)
        save=QPushButton('保存',self)
        save.clicked.connect(self.save_picture)
        return label1,self.spin,label2,self.color_change,save
    def get_widget2(self):
        label=QLabel('迭代次数',self)
        self.iters=QSpinBox(self)
        self.iters.setValue(10)
        mode=QLabel('模式',self)
        self.nums=QSpinBox(self)
        self.nums.setValue(1)
        self.nums.setMinimum(1)
        self.nums.setMaximum(2)
        white=QPushButton('白色',self)
        white.clicked.connect(self.white_pic)
        blue=QPushButton('蓝色',self)
        blue.clicked.connect(self.white_pic)
        return label,self.iters,mode,self.nums,white,blue
    def get_iters(self):
        return self.iters.value()
    def get_nums(self):
        return self.nums.value()
    def imread(self,path):
# 解决中文路径问题
        cv_img = cv.imdecode(np.fromfile(path,dtype=np.uint8),-1)
        return cv_img
    def white_pic(self):
        text = self.sender().text()
        self.picture = self.imread(self.filename)
        tool = change_color(self.picture)
        tool.get_color(self.color_value())
        tool.set_iter(self.get_iters())
        tool.set_num(self.get_nums())
        tool.main(text)
        img = tool.return_picture()
        h, w, _ = img.shape
        img = QImage(img.data, w, h,w*3,QImage.Format_RGB888)
        img=QPixmap.fromImage(img)
        self.board.get_picture(img)
    def main(self,what):
# switch的选择，与可以用if...else..
        match what:
            case '白色':
                self.start(self.white)
            case '蓝色':
                self.start(self.blue)

    def save_picture(self):
        path,ok = QFileDialog.getSaveFileName(self, '保存图片', '.\\', '(*.jpg *.png)')
        if ok:
            img=self.board.save()
            img.save(path)

    def color_value(self):
        color=QColorDialog.getColor()
        self.board.get_color(color)
        color=color.getRgb()[:-1][::-1]
        return color
    def spin_value(self):
        self.size=self.spin.value()
        self.board.change_spin(self.size)
    def get_paint(self):
        image=QPixmap(self.filename)
        self.board=cv_Board(image,self.size)
        return self.board
class Start(QWidget):
# 开始页面
    def __init__(self):
        super().__init__()
        self.filename=None
        self.main=None
        self.initui()
    def initui(self):
        self.resize(640,480)
        self.setWindowTitle('主页面')
        self.set_layout()
    def set_layout(self):
        v1=QHBoxLayout(self)
        open_btn=self.get_btn()
        v1.addWidget(open_btn)
        return v1
    def get_btn(self):
        btn1=QPushButton('打开图片',self)
        btn1.clicked.connect(self.open)
        return btn1
    def open(self):
        filename, ok = QFileDialog.getOpenFileName(self, '打开文件', os.getcwd(), '(*.jpg *.png)')
        if ok:
            self.filename=filename
            self.main=Main(self.filename)
            self.main.setWindowModality(Qt.ApplicationModal)
            self.main.showMaximized()
    def paintEvent(self, a0):
        paint = QPainter(self)
        pixmap = QPixmap('C:/Users/520/PycharmProjects/pythonProject4/图片/图片/29.jpg')
        paint.drawPixmap(self.rect(), pixmap)



if __name__ == '__main__':
    app = QApplication(sys.argv)
    a = Start()
    a.show()
    app.exec_()

代码比较多，不想写注释了，直接演示一下把

操作

蓝色

运行，就只有一个按钮，先要打开图片

先来简单的，比如打开这张图片（这张图片是网上随便找的）

打开后

就会弹出

布局确实不好看。。。。（0.0）

可以修改迭代次数，默认为10，比如设为1

而且要用开运算，所以模式还要设为2

因为底色是蓝色的，所以点击蓝色的按钮，然后弹出颜色选择框，不如选择白色的

结果

可以看到，效果还是很好的，主要是大佬的二值化范围设定的好，然后可以选择保存。

白色

来张白色背景的照片（网上搜的）

同理，运行到这一步

经过多次对参数的修改以及，范围的修改

最终得到的图片是这样的，可以看到有白色的，很由可能是我对白色的选取不到位。。。

范围不好说，然后接下来进行人工处理

当然，手抖，确实有问题，修的不是很好

总结

对白色底色的范围选取，并不是很完美，有待实验

底色可以任意选择，黄蓝绿等

看

有瑕疵。。