import cv2
import matplotlib.pyplot as plt

# 讀取圖片，使用灰度模式讀取圖片
image = cv2.imread('aaa.png', cv2.IMREAD_GRAYSCALE)

# 使用Canny邊緣檢測(cè)算法進(jìn)行邊緣提取，threshold1和threshold2是邊緣檢測(cè)的閾值
edges = cv2.Canny(image, threshold1=100, threshold2=200)

# 創(chuàng)建一個(gè)圖形窗口，設(shè)置顯示圖像的大小
plt.figure(figsize=(10, 5))

# 顯示原始灰度圖像
plt.subplot(1, 2, 1)  # 將顯示區(qū)域分為1行2列，當(dāng)前顯示第一列
plt.title('Original Image')  # 給當(dāng)前圖像設(shè)置標(biāo)題
plt.imshow(image, cmap='gray')  # 顯示圖片，使用灰度色圖

# 顯示經(jīng)過(guò)Canny算法處理后的邊緣圖像
plt.subplot(1, 2, 2)  # 當(dāng)前顯示第二列
plt.title('Edges')  # 給邊緣圖像設(shè)置標(biāo)題
plt.imshow(edges, cmap='gray')  # 顯示邊緣檢測(cè)結(jié)果，使用灰度色圖

# 顯示所有圖像
plt.show()

# 將邊緣檢測(cè)的結(jié)果保存為圖片
cv2.imwrite('edges_output.jpg', edges)

import sys
import cv2
import os
from PyQt5.QtWidgets import QApplication, QWidget, QVBoxLayout, QPushButton, QLabel, QFileDialog, QMessageBox
from PyQt5.QtCore import Qt


class ImageProcessor(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        self.setWindowTitle('批量圖像邊緣檢測(cè)')
        self.setGeometry(900, 500, 400, 200)

        layout = QVBoxLayout()

        self.label = QLabel('選擇圖片進(jìn)行批量邊緣檢測(cè)', self)
        self.label.setAlignment(Qt.AlignCenter)
        layout.addWidget(self.label)

        self.btn_select = QPushButton('選擇圖片', self)
        self.btn_select.clicked.connect(self.select_images)
        layout.addWidget(self.btn_select)

        self.btn_process = QPushButton('處理并保存', self)
        self.btn_process.clicked.connect(self.process_images)
        layout.addWidget(self.btn_process)

        self.setLayout(layout)

    def select_images(self):
        # 選擇多個(gè)圖片文件
        options = QFileDialog.Options()
        self.filenames, _ = QFileDialog.getOpenFileNames(self, "選擇圖片文件", "",
                                                         "Images (*.png *.jpg *.bmp);;All Files (*)", options=options)
        if self.filenames:
            # 獲取當(dāng)前工作目錄或自定義目錄作為基準(zhǔn)目錄
            base_dir = os.getcwd()  # 或者指定一個(gè)路徑，例如 r"C:\Users\小莊的Y9000P\Desktop"

            # 將絕對(duì)路徑轉(zhuǎn)換為相對(duì)路徑
            self.filenames = [os.path.relpath(filename, base_dir) for filename in self.filenames]

            self.label.setText(f'已選擇 {len(self.filenames)} 張圖片')

    def process_images(self):
        if not self.filenames:
            QMessageBox.warning(self, '警告', '請(qǐng)先選擇圖片文件')
            return

        for filename in self.filenames:
            try:
                print(f"正在處理文件: {filename}")  # 添加調(diào)試信息
                # 讀取圖片，使用灰度模式讀取圖片
                image = cv2.imread(filename, cv2.IMREAD_GRAYSCALE)

                if image is None:
                    print(f"無(wú)法讀取文件: {filename}")  # 添加調(diào)試信息
                    raise Exception(f"無(wú)法讀取文件: {filename}")

                # 使用Canny邊緣檢測(cè)算法進(jìn)行邊緣提取
                edges = cv2.Canny(image, threshold1=100, threshold2=200)

                # 保存邊緣檢測(cè)結(jié)果
                output_filename = filename.rsplit('.', 1)[0] + '_edges.jpg'
                cv2.imwrite(output_filename, edges)

            except Exception as e:
                QMessageBox.warning(self, '錯(cuò)誤', f'處理文件 {filename} 時(shí)出錯(cuò): {str(e)}')
                continue

        QMessageBox.information(self, '完成', '所有圖片處理完成并已保存')


if __name__ == '__main__':
    app = QApplication(sys.argv)
    ex = ImageProcessor()
    ex.show()
    sys.exit(app.exec_())