本文共 4914 字，大约阅读时间需要 16 分钟。

   
   
   
import cv2
import numpy as np
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtWidgets import QApplication, QMainWindow, QMessageBox, QFileDialog
class Ui_MainWindow(object):
    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(800, 600)
        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")
        self.pushButton = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton.setGeometry(QtCore.QRect(280, 0, 251, 331))
        self.pushButton.setObjectName("pushButton")
        self.pushButton.clicked.connect(self.openFile)
        self.pushButton_2 = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton_2.setGeometry(QtCore.QRect(140, 390, 101, 51))
        self.pushButton_2.setObjectName("pushButton_2")
        self.pushButton_2.clicked.connect(lambda: self.setimg_bg("blue"))
        self.pushButton_3 = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton_3.setGeometry(QtCore.QRect(280, 390, 101, 51))
        self.pushButton_3.setObjectName("pushButton_3")
        self.pushButton_3.clicked.connect(lambda: self.setimg_bg("green"))
        self.pushButton_4 = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton_4.setGeometry(QtCore.QRect(430, 390, 101, 51))
        self.pushButton_4.setObjectName("pushButton_4")
        self.pushButton_4.clicked.connect(lambda: self.setimg_bg("red"))
        self.pushButton_5 = QtWidgets.QPushButton(self.centralwidget)
        self.pushButton_5.setGeometry(QtCore.QRect(570, 390, 101, 51))
        self.pushButton_5.setObjectName("pushButton_5")
        self.pushButton_5.clicked.connect(lambda: self.setimg_bg("white"))
        MainWindow.setCentralWidget(self.centralwidget)
        self.menubar = QtWidgets.QMenuBar(MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 26))
        self.menubar.setObjectName("menubar")
        MainWindow.setMenuBar(self.menubar)
        self.statusbar = QtWidgets.QStatusBar(MainWindow)
        self.statusbar.setObjectName("statusbar")
        MainWindow.setStatusBar(self.statusbar)
        self.retranslateUi(MainWindow)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)
    def openFile(self):
        global img_path
        get_filename_path, ok = QFileDialog.getOpenFileName()
        if ok:
            img_path = str(get_filename_path)
            self.pushButton.setStyleSheet("QPushButton{border-image: url(%s)}" % str(get_filename_path))
    def setimg_bg(self, color):
        global img_path, new_path
        if img_path != '':
            change_bg_color(img_path, color)
            self.pushButton.setStyleSheet("QPushButton{border-image: url(%s)}" % str(new_path))
    def retranslateUi(self, MainWindow):
        _translate = QtCore.QCoreApplication.translate
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
        self.pushButton.setText(_translate("MainWindow", ""))
        self.pushButton_2.setText(_translate("MainWindow", "蓝色"))
        self.pushButton_3.setText(_translate("MainWindow", "绿色"))
        self.pushButton_4.setText(_translate("MainWindow", "红色"))
        self.pushButton_5.setText(_translate("MainWindow", "白色"))
def change_bg_color(path, color):
    color_dict = {
        'red': [0, 0, 255],
        'green': [0, 255, 0],
        'blue': [255, 0, 0],
        'white': [255, 255, 255]
    }
    color_list = color_dict[color]
    img = cv2.imdecode(np.fromfile(path, dtype=np.uint8), 1)
    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    if img[0][0][0] > 200 and img[0][0][1] < 230 and img[0][0][2] < 230:
        hsv_min = np.array([47, 79, 79])
        hsv_max = np.array([102, 255, 255])
    elif img[0][0][1] > 200 and img[0][0][0] < 230 and img[0][0][2] < 230:
        hsv_min = np.array([41, 40, 41])
        hsv_max = np.array([90, 255, 255])
    elif img[0][0][2] > 200 and img[0][0][0] < 230 and img[0][0][1] < 230:
        hsv_min = np.array([0, 200, 40])
        hsv_max = np.array([10, 255, 255])
    else:
        hsv_min = np.array([0, 0, 221])
        hsv_max = np.array([180, 30, 255])
    mask = cv2.inRange(hsv, hsv_min, hsv_max)
    erode = cv2.erode(mask, None, iterations=1)
    dilate = cv2.dilate(erode, None, iterations=1)
    rows, cols, channels = img.shape
    for i in range(rows):
        for j in range(cols):
            if dilate[i, j] == 255:
                img[i, j] = (color_list[0], color_list[1], color_list[2])
    new_path = f'%s_%s.jpg' % (str(int(time.time())), color)
    path = path.replace(path.split('/')[-1], new_path)
    cv2.imencode('.jpg', img)[1].tofile(path)
if __name__ == '__main__':
    global img_path, new_path
    img_path = 'G:/34.jpg'
    app = QApplication(sys.argv)
    MainWindow = QMainWindow()
    ui = Ui_MainWindow()
    ui.setupUi(MainWindow)
    MainWindow.show()
    sys.exit(app.exec_())
   
   

转载地址：http://mwvp.baihongyu.com/