[發問] 新手問題 python3 & OpenCV 4.5.3 版本 關於執行程式會卡頓的問題

軒云熊

這是一個在照片上畫直線 然後量測值線長度的程式練習
這段程式代碼有一個問題 在執行程式的時候 會卡頓
在想應該是 迴圈的問題  函式中 有一段 :

        directory = 'C:/VSC_PY/GUI_AI/sand_train/text_dir/'  # 目錄路徑
        total_items = count_files(directory)
        if total_items > 1:
            print("此目錄下的項數總計大於 2。")
            print(total_items)
            # 讀取照片
            img1 = cv2.imread('C:/VSC_PY/GUI_AI/sand_train/text_dir/TEST1.jpg')
            #複製照片避免殘影現象 None   
            img1 = img1.copy()  
        else:
            print("此目錄下的項數總計不足 2。")
            img1 = img.copy()

應該是這一段造成的卡頓 不知道 有沒有大大可以告知小弟
該如何修改才不會卡頓

1. #!/usr/bin/env python3
   
2. # -*- coding: utf-8 -*-
   
3. import cv2 #OpenCV version: 4.5.3  print("OpenCV version:", cv2.__version__)
   
4. # import numpy as np
   
5. import math
   
6. import os
   
7. 
   
8. # 列出指定目錄中的所有檔案和子目錄。
   
9. def count_files(directory):
   
10.     total = 0
    
11.     for root, dirs, files in os.walk(directory):
    
12.         total += len(files) + len(dirs)
    
13.     return total
    
14. 
    
15. # 刪除的圖像
    
16. def delete_image(directory, filename):
    
17.     filepath = os.path.join(directory, filename)
    
18.     if os.path.isfile(filepath):
    
19.         os.remove(filepath)
    
20.         print(f"Deleted {filename}.")
    
21.     else:
    
22.         print(f"{filename} 不存在")
    
23. 
    
24. # 當左鍵按下並移動時繪製圖形，event可以查看移動，flag查看是否按下
    
25. # 定義 `draw_line()` 函數來處理滑鼠事件，例如在圖像上繪製線條
    
26. def draw_line(event, x, y, flags, param):
    
27.     if event == cv2.EVENT_FLAG_LBUTTON:  # 左鍵按下
    
28.         global start_point, isDrawing ,img1
    
29. 
    
30.         start_point = (x, y) # 拖曳滑鼠時，不斷記錄座標
    
31.         isDrawing = True
    
32.         print("左鍵按下")
    
33.         print(start_point)
    
34. 
    
35.     elif event == cv2.EVENT_MOUSEMOVE and isDrawing:  # 移動滑鼠時繪圖
    
36. 
    
37.         # end_point = (x,y) # 拖曳滑鼠時，不斷記錄座標
    
38.         end_point = (x,y)
    
39.         print("移動滑鼠")
    
40.         print(end_point)
    
41. 
    
42.         directory = 'C:/VSC_PY/GUI_AI/sand_train/text_dir/'  # 目錄路徑
    
43.         total_items = count_files(directory)
    
44.         if total_items > 1:
    
45.             print("此目錄下的項數總計大於 2。")
    
46.             print(total_items)
    
47.             # 讀取照片
    
48.             img1 = cv2.imread('C:/VSC_PY/GUI_AI/sand_train/text_dir/TEST1.jpg')
    
49.             #複製照片避免殘影現象 None   
    
50.             img1 = img1.copy()  
    
51.         else:
    
52.             print("此目錄下的項數總計不足 2。")
    
53.             img1 = img.copy()
    
54.             
    
55. 
    
56.         # P1 與 P2 是已知的點
    
57.         # P1 = (start_point[0], start_point[1])
    
58.         # P2 = (end_point[0], end_point[1])  
    
59.         P1 = (start_point[-2], start_point[-1])
    
60.         P2 = (end_point[-2], end_point[-1])  
    
61.         cv2.line(img1, P1, P2, (0, 255, 0), 3)
    
62. 
    
63.         # 計算直線長度公式為：length = sqrt((x2-x1)^2 + (y2-y1)^2)
    
64.         length = math.sqrt((P2[0] - P1[0])**2 + (P2[1] - P1[1])**2)
    
65.         # length = np.sqrt((P2[0]-P1[0])**2 + (P2[1]-P1[1])**2) # 計算直線的長度
    
66.         # 長度單位 micrometer（um）
    
67.         length = length * 1000
    
68. 
    
69.         # 顯示座標  cv2.threshold() strXY=str(x)+','+str(y)+"um"
    
70.         strXY=str(x)+','+str(y)+','+str(length)+"um"
    
71.         font=cv2.FONT_HERSHEY_PLAIN
    
72.         cv2.putText(img1,strXY,(x+10,y+10),font,1,(0,255,0),2)
    
73. 
    
74.         cv2.imshow("Image", img1)
    
75. 
    
76.     elif event == cv2.EVENT_LBUTTONUP:  # 左鍵釋放 True False
    
77.         isDrawing = False
    
78.         print("左鍵釋放")
    
79.         print(isDrawing)
    
80.         #保存照片
    
81.         cv2.imwrite('C:/VSC_PY/GUI_AI/sand_train/text_dir/TEST1.jpg', img1)
    
82.    
    
83.     elif event == cv2.EVENT_RBUTTONDOWN:
    
84.         print("右鍵按下")
    
85. 
    
86.         if __name__ == "__main__":
    
87.             directory = 'C:/VSC_PY/GUI_AI/sand_train/text_dir/'  # 目錄路徑
    
88.             filename = "TEST1.jpg" # 圖片的名稱
    
89.             delete_image(directory, filename) #刪除照片
    
90.         img3 = img.copy()
    
91.         cv2.imshow("Image", img3)     
    
92. 
    
93. # 載入圖片 'C:/VSC_PY/GUI_AI/sand_train/text_dir/TEST.jpg'
    
94. # 預設使用 cv2.IMREAD_COLOR 模式
    
95. img = cv2.imread('C:/VSC_PY/GUI_AI/sand_train/text_dir/TEST.jpg')
    
96. 
    
97. # cv2.imshow('oxxostudio', img) # 使用名為 oxxostudio 的視窗開啟圖片
    
98. 
    
99. # 建立一個視窗以顯示影像並設定事件監聽器
    
100. cv2.namedWindow("Image")
     
101. # 偵測指定視窗下的滑鼠事件
     
102. cv2.setMouseCallback("Image", draw_line)
     
103. 
     
104. # 啟動主迴圈以顯示和更新圖像，包括繪製的線條
     
105. while True:
     
106.     cv2.imshow("Image", img)      
     
107.     # 按下任意鍵停止
     
108.     if cv2.waitKey(0):
     
109.         cv2.destroyAllWindows() # 結束所有圖片視窗
     
110.         break

複製代碼