75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
import cv2
|
|
import numpy as np
|
|
import os
|
|
from tkinter import Tk, filedialog
|
|
|
|
def select_image():
|
|
Tk().withdraw() # 隱藏主視窗
|
|
file_path = filedialog.askopenfilename(
|
|
title="選擇影像檔案",
|
|
filetypes=[("影像檔案", "*.png;*.jpg;*.jpeg;*.bmp")]
|
|
)
|
|
return file_path
|
|
|
|
def read_image_with_unicode_path(image_path):
|
|
with open(image_path, 'rb') as file:
|
|
binary_data = np.asarray(bytearray(file.read()), dtype=np.uint8)
|
|
image = cv2.imdecode(binary_data, cv2.IMREAD_COLOR)
|
|
return image
|
|
|
|
def save_image(output_path, image):
|
|
# 儲存影像到指定路徑
|
|
cv2.imwrite(output_path, image)
|
|
print(f"影像已儲存至：{output_path}")
|
|
|
|
def detect_lines_with_hough(image):
|
|
# 二值化處理
|
|
_, binary_image = cv2.threshold(image, 68, 255, cv2.THRESH_BINARY)
|
|
|
|
# 儲存二值化影像
|
|
binary_image_path = os.path.join(r"E:\AP", f"binary.png")
|
|
save_image(binary_image_path, binary_image)
|
|
|
|
# 霍夫直線偵測
|
|
lines = cv2.HoughLinesP(binary_image, 1, np.pi / 180, threshold=100, minLineLength=1300, maxLineGap=50)
|
|
|
|
# 在原影像上繪製直線
|
|
line_image = cv2.cvtColor(image, cv2.COLOR_GRAY2BGR) # 將灰階影像轉為 BGR 以繪製彩色直線
|
|
if lines is not None:
|
|
for line in lines:
|
|
x1, y1, x2, y2 = line[0]
|
|
cv2.line(line_image, (x1, y1), (x2, y2), (0, 0, 255), 2)
|
|
|
|
return line_image
|
|
|
|
def process_image(image_path):
|
|
# 讀取影像
|
|
image = read_image_with_unicode_path(image_path)
|
|
|
|
if image is None:
|
|
print("無法讀取影像")
|
|
return
|
|
|
|
# 轉換為灰階影像
|
|
gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
|
|
|
|
# 執行霍夫直線偵測（前處理為二值化）
|
|
line_image = detect_lines_with_hough(gray_image)
|
|
|
|
# 取得輸出路徑
|
|
directory = os.path.dirname(r"E:\AP\img0120")
|
|
|
|
gray_image_path = os.path.join(directory, f"gray.png")
|
|
lines_image_path = os.path.join(directory, f"lines.png")
|
|
|
|
# 儲存影像
|
|
save_image(gray_image_path, gray_image)
|
|
save_image(lines_image_path, line_image)
|
|
|
|
if __name__ == "__main__":
|
|
file_path = select_image()
|
|
if file_path:
|
|
process_image(file_path)
|
|
else:
|
|
print("未選擇影像檔案")
|