228 lines
9.0 KiB
Python
228 lines
9.0 KiB
Python
import sys
|
||
from PyQt5 import QtWidgets, QtGui, QtCore
|
||
from PyQt5.QtCore import QTimer
|
||
from PyQt5.QtWidgets import QFileDialog
|
||
from pypylon import pylon
|
||
import cv2
|
||
import numpy as np
|
||
import os
|
||
from Main import Ui_MainWindow
|
||
|
||
class CameraApp(QtWidgets.QMainWindow, Ui_MainWindow):
|
||
def __init__(self):
|
||
super(CameraApp, self).__init__()
|
||
self.setupUi(self)
|
||
|
||
# 連接按鈕事件
|
||
self.bt_camera_connect.clicked.connect(self.connect_camera)
|
||
self.bt_OneShot.clicked.connect(self.one_shot_capture)
|
||
self.bt_KeetShot.clicked.connect(self.keep_shot_capture)
|
||
self.bt_open_image.clicked.connect(self.open_image)
|
||
self.bt_save_image.clicked.connect(self.save_image) # 連接儲存影像按鈕事件
|
||
|
||
|
||
# 初始化變數
|
||
self.camera = None
|
||
self.timer = QTimer()
|
||
self.timer.timeout.connect(self.keep_shot_process)
|
||
|
||
# 設定 QSlider 範圍和事件
|
||
self.threshold_value.setMinimum(0)
|
||
self.threshold_value.setMaximum(255)
|
||
self.threshold_value.setValue(127) # 預設閥值
|
||
self.threshold_value.valueChanged.connect(self.update_threshold)
|
||
|
||
self.threshold.setText(str(self.threshold_value.value()))
|
||
self.current_image = None
|
||
|
||
def get_bgr_image(self):
|
||
"""取得 BGR 格式的影像"""
|
||
if self.current_image is None:
|
||
return None
|
||
|
||
# 若影像為單通道或 Bayer 格式，進行轉換
|
||
if len(self.current_image.shape) == 2 or self.current_image.shape[2] == 1:
|
||
return cv2.cvtColor(self.current_image, cv2.COLOR_BayerBG2BGR)
|
||
else:
|
||
return self.current_image
|
||
|
||
def connect_camera(self):
|
||
try:
|
||
self.camera = pylon.InstantCamera(pylon.TlFactory.GetInstance().CreateFirstDevice())
|
||
self.camera.Open()
|
||
self.statusbar.showMessage("相機連線成功")
|
||
except Exception as e:
|
||
self.statusbar.showMessage(f"相機連線失敗: {e}")
|
||
|
||
def get_exposure_time(self):
|
||
"""從 QTextEdit 取得曝光時間，若為空則使用預設值"""
|
||
exposure_time_text = self.Ex_time.toPlainText().strip()
|
||
|
||
# 當 QTextEdit 值為空時使用預設值
|
||
if not exposure_time_text:
|
||
return 5000 # 預設曝光時間（微秒）
|
||
|
||
# 驗證是否為數值
|
||
if exposure_time_text.isdigit():
|
||
return int(exposure_time_text)
|
||
|
||
self.statusbar.showMessage("請輸入有效的曝光時間（整數），使用預設值 5000 微秒")
|
||
return 5000
|
||
|
||
def one_shot_capture(self):
|
||
if not (self.camera and self.camera.IsOpen()):
|
||
self.statusbar.showMessage("請先連接相機")
|
||
return
|
||
|
||
try:
|
||
exposure_time = self.get_exposure_time()
|
||
self.camera.ExposureTime.SetValue(float(exposure_time))
|
||
self.statusbar.showMessage(f"曝光時間設定為 {exposure_time} 微秒")
|
||
|
||
if self.camera.IsGrabbing():
|
||
self.camera.StopGrabbing()
|
||
|
||
self.camera.StartGrabbing(1)
|
||
grab_result = self.camera.RetrieveResult(5000, pylon.TimeoutHandling_Return)
|
||
|
||
if grab_result.GrabSucceeded():
|
||
self.current_image = grab_result.Array
|
||
self.display_original_image()
|
||
self.apply_threshold_and_display()
|
||
self.statusbar.showMessage("單張擷取成功")
|
||
else:
|
||
self.statusbar.showMessage("擷取失敗")
|
||
|
||
grab_result.Release()
|
||
|
||
except Exception as e:
|
||
self.statusbar.showMessage(f"擷取失敗: {e}")
|
||
if self.camera.IsGrabbing():
|
||
self.camera.StopGrabbing()
|
||
|
||
def display_original_image(self):
|
||
"""顯示原始影像到 label 上"""
|
||
if self.current_image is not None:
|
||
# 判斷是否需要轉換 Bayer 格式
|
||
if len(self.current_image.shape) == 2 or self.current_image.shape[2] == 1:
|
||
# 影像來自相機（Bayer 格式）
|
||
image_bgr = cv2.cvtColor(self.current_image, cv2.COLOR_BayerBG2BGR)
|
||
else:
|
||
# 影像已經是 BGR 格式
|
||
image_bgr = self.current_image
|
||
|
||
# 顯示影像
|
||
height, width, channel = image_bgr.shape
|
||
bytes_per_line = 3 * width
|
||
qimage = QtGui.QImage(image_bgr.data, width, height, bytes_per_line, QtGui.QImage.Format_BGR888)
|
||
pixmap = QtGui.QPixmap.fromImage(qimage).scaled(self.label.size(), QtCore.Qt.KeepAspectRatio)
|
||
self.label.setPixmap(pixmap)
|
||
|
||
def apply_threshold_and_display(self):
|
||
"""套用二值化處理並顯示影像到 label_2 上"""
|
||
if self.current_image is not None:
|
||
# 判斷是否需要轉換 Bayer 格式
|
||
if len(self.current_image.shape) == 2 or self.current_image.shape[2] == 1:
|
||
# 影像來自相機（Bayer 格式）
|
||
image_bgr = cv2.cvtColor(self.current_image, cv2.COLOR_BayerBG2BGR)
|
||
else:
|
||
# 影像已經是 BGR 格式
|
||
image_bgr = self.current_image
|
||
|
||
# 轉換為灰階
|
||
image_gray = cv2.cvtColor(image_bgr, cv2.COLOR_BGR2GRAY)
|
||
|
||
# 套用二值化處理
|
||
threshold_value = self.threshold_value.value()
|
||
_, thresholded_image = cv2.threshold(image_gray, threshold_value, 255, cv2.THRESH_BINARY)
|
||
|
||
# 顯示影像
|
||
height, width = thresholded_image.shape
|
||
bytes_per_line = width
|
||
qimage = QtGui.QImage(thresholded_image.data, width, height, bytes_per_line, QtGui.QImage.Format_Grayscale8)
|
||
pixmap = QtGui.QPixmap.fromImage(qimage).scaled(self.label_2.size(), QtCore.Qt.KeepAspectRatio)
|
||
self.label_2.setPixmap(pixmap)
|
||
|
||
def update_threshold(self):
|
||
"""當 QSlider 值改變時更新影像"""
|
||
self.apply_threshold_and_display()
|
||
# 更新 QTextEdit 顯示的值
|
||
self.threshold.setText(str(self.threshold_value.value()))
|
||
|
||
def keep_shot_capture(self):
|
||
if self.camera and self.camera.IsOpen():
|
||
if not self.timer.isActive():
|
||
self.camera.StartGrabbing(pylon.GrabStrategy_LatestImageOnly)
|
||
self.timer.start(30)
|
||
self.statusbar.showMessage("開始連續取像")
|
||
else:
|
||
self.timer.stop()
|
||
self.camera.StopGrabbing()
|
||
self.statusbar.showMessage("停止連續取像")
|
||
else:
|
||
self.statusbar.showMessage("請先連接相機")
|
||
|
||
def keep_shot_process(self):
|
||
if self.camera and self.camera.IsGrabbing():
|
||
grab_result = self.camera.RetrieveResult(5000, pylon.TimeoutHandling_ThrowException)
|
||
if grab_result.GrabSucceeded():
|
||
self.current_image = grab_result.Array
|
||
self.display_original_image()
|
||
self.apply_threshold_and_display()
|
||
grab_result.Release()
|
||
|
||
def open_image(self):
|
||
options = QFileDialog.Options()
|
||
file_path, _ = QFileDialog.getOpenFileName(self, "打開影像檔", "", "Images (*.png *.xpm *.jpg *.bmp *.tiff)", options=options)
|
||
|
||
if file_path:
|
||
if not os.path.exists(file_path):
|
||
self.statusbar.showMessage("檔案不存在或路徑錯誤")
|
||
return
|
||
|
||
image_data = np.fromfile(file_path, dtype=np.uint8)
|
||
image = cv2.imdecode(image_data, cv2.IMREAD_COLOR)
|
||
|
||
if image is not None:
|
||
self.current_image = image
|
||
self.display_original_image()
|
||
self.apply_threshold_and_display()
|
||
self.statusbar.showMessage("影像載入成功")
|
||
else:
|
||
self.statusbar.showMessage("無法讀取影像")
|
||
def save_image(self):
|
||
"""將二值化影像儲存到檔案"""
|
||
if self.current_image is None:
|
||
self.statusbar.showMessage("沒有影像可儲存")
|
||
return
|
||
|
||
# 使用 QFileDialog 取得儲存檔案路徑和名稱
|
||
file_path, _ = QFileDialog.getSaveFileName(self, "儲存二值化影像", "", "Images (*.png *.jpg *.bmp *.tiff)")
|
||
|
||
if file_path:
|
||
try:
|
||
# 取得 BGR 影像並生成二值化影像
|
||
image_bgr = self.get_bgr_image()
|
||
if image_bgr is None:
|
||
self.statusbar.showMessage("無法取得影像")
|
||
return
|
||
|
||
image_gray = cv2.cvtColor(image_bgr, cv2.COLOR_BGR2GRAY)
|
||
threshold_value = self.threshold_value.value()
|
||
_, thresholded_image = cv2.threshold(image_gray, threshold_value, 255, cv2.THRESH_BINARY)
|
||
|
||
# 儲存影像
|
||
cv2.imwrite(file_path, thresholded_image)
|
||
self.statusbar.showMessage(f"二值化影像成功儲存到 {file_path}")
|
||
|
||
except Exception as e:
|
||
self.statusbar.showMessage(f"影像儲存失敗: {e}")
|
||
|
||
|
||
|
||
if __name__ == "__main__":
|
||
app = QtWidgets.QApplication(sys.argv)
|
||
window = CameraApp()
|
||
window.show()
|
||
sys.exit(app.exec_())
|