Python相片更换背景颜色qt窗体程序

Python相片更换背景颜色qt窗体程序

如需远程调试，可加QQ905733049由专业技术人员远程协助！

运行代码如下：

# -*- coding: utf-8 -*-
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5 import QtWidgets
from PyQt5.QtWidgets import *
import cv2 as cv
import numpy as np
import os


class MainWindow(QMainWindow, Ui_MainWindow):
    """
    Class documentation goes here.
    """
    def __init__(self, parent=None):
        """
        Constructor
        
        @param parent reference to the parent widget
        @type QWidget
        """
        super(MainWindow, self).__init__(parent)
        self.setupUi(self)
        self.color = [0, 0, 0]
        self.target =True 
        self.i=0
        self.newname =''
        self.file_name=""
        self.file_type =""
    
    @pyqtSlot()
    def on_loadImage_clicked(self):
        """
        Slot documentation goes here.
        """
        # TODO: not implemented yet
        
        self.file_name, self.file_type = QFileDialog.getOpenFileName(self,"打开文件","D:/","All Files (*);")

        if (img.shape[2] == 3):#bgr到rgb转换的显示
            qimg = cv.cvtColor(img, cv.COLOR_BGR2RGB)
            qimg = QImage(qimg.data, qimg.shape[1], qimg.shape[0], qimg.shape[1]*qimg.shape[2],QImage.Format_RGB888)
            self.before.setPixmap(QPixmap.fromImage(qimg))
        else:
            qimg = QImage(img.data, img.shape[1], img.shape[0],QImage.Format_Grayscale8)
            self.before.setPixmap(QPixmap.fromImage(qimg))
    
        

    
    @pyqtSlot(str)
    def on_colorchoose_currentIndexChanged(self, p0):
        """
        Slot documentation goes here.
        
        @param p0 DESCRIPTION
        @type str
        """
        # TODO: not implemented yet
        current = self.colorchoose.currentText()
        if self.target:
            if current=="红色":
                self.color =[0, 0, 255]
                self.newname= '_red'
            elif current=="白色":
                self.color =[255, 255, 255]
                self.newname= '_white'
            elif current=="蓝色":
                self.color =[255, 0, 0]
                self.newname= '_blue'
            elif current=="绿色":
                self.color =[0, 255, 0]
                self.newname= '_green'
            elif current=="黄色":
                self.color=[0, 255, 255]
                self.newname= '_yellow'
            elif current=="紫色":
                self.color=[255, 0, 255]
                self.newname= '_violet'
            elif current=="灰色":
                self.color=[96, 96, 96]
                self.newname= '_gray'
    
    @pyqtSlot()
    def on_change_clicked(self):
        """
        Slot documentation goes here.
        """
        # TODO: not implemented yet
        if self.red.text()!="":
            self.newname='_define'
            print(self.color)
            self.target = False
        img =cv.imread(self.file_name, -1)

        if (img is None):
            print(1)
            self.textBrowser.setText("打开图片失败，请重新加载")
        else:
            # 准备数据
            wide = img.shape[1]
            height = img.shape[0]
            dims = img.shape[2]
            sampleCount = wide * height
            clusterCount = 4
            points = np.zeros([sampleCount, dims], np.float32)
            centers = np.zeros([clusterCount, 1], np.float32)
            index = 0
            for i in range(height):
                for j in range(wide):
                    index = i * wide + j
                    for n in range(dims):
                        points[index][n] = img[i][j][n]
            criteria = (cv.TERM_CRITERIA_EPS + cv.TERM_CRITERIA_MAX_ITER, 10, 0.1)
            ret, label, center = cv.kmeans(points, clusterCount, None, criteria, 10, cv.KMEANS_PP_CENTERS)
            mask = np.zeros((img.shape[0], img.shape[1]), np.uint8)
            index = wide * 2 + 4
            img2 = img.copy()
            cindex = label[index]
            for i in range(height):
                for j in range(wide):
                    index = i * wide + j
                    label1 = label[index]
                    if cindex == label1:
                        for n in range(dims):
                            img2[i][j][n] = 0;
                        mask[i][j] = 0;
                    else:
                        mask[i][j] = 255
            structuting = cv.getStructuringElement(cv.MORPH_RECT, (13, 13))
            mask =cv.dilate(mask,structuting)
            mask = cv.erode(mask, structuting)
            structuting2 = cv.getStructuringElement(cv.MORPH_RECT, (9, 9))
            mask = cv.erode(mask, structuting2)
            cv.waitKey(10)
            mask = cv.GaussianBlur(mask, (9, 9), 0, 0)
            for i in range(height):
                for j in range(wide):
                    if mask[i][j] == 0:
                        for n in range(dims):
                            img2[i][j][n] =  self.color[n]
                    else:
                        continue
            img2 = cv.GaussianBlur(img2, (3, 3), 0, 0)
            if (img2.shape[2] == 3):
                qimg = cv.cvtColor(img2, cv.COLOR_BGR2RGB)
                qimg = QImage(qimg.data, qimg.shape[1], qimg.shape[0], qimg.shape[1] * qimg.shape[2],
                              QImage.Format_RGB888)
                self.after.setPixmap(QPixmap.fromImage(qimg))
            else:
                qimg = QImage(img.data, img.shape[1], img.shape[0], QImage.Format_Grayscale8)
                self.after.setPixmap(QPixmap.fromImage(qimg))
            name=fathername+"/"+name+self.newname+type
            print(name)
            cv.imwrite(name,img2)
            write = "the image has been saved in:"+name
            self.textBrowser.setText(write)

        
if __name__ == "__main__":
    import sys
    app = QtWidgets.QApplication(sys.argv)
    ui = MainWindow()
    ui.show()
    sys.exit(app.exec_())
    

运行结果：