0601

工程/Area_tracking.py

@@ -1,72 +1,84 @@
 import cv2
 import numpy as np
 
-def init_camera():
+def main():
-    global cap
+    # 初始化摄像头
     cap = cv2.VideoCapture(0)
 
-def select_color_area(frame):
+    # 用于标记是否已经选择颜色区域的变量
-    global color_selected, greenLower, greenUpper
-    cv2.imshow('Select color area', frame)
-
-    if cv2.waitKey(1) & 0xFF == ord('c'):
-        hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
-        roi = cv2.selectROI('Select color area', hsv)
-        if roi != (0, 0, 0, 0):
-            roi_hsv = hsv[int(roi[1]):int(roi[1] + roi[3]), int(roi[0]):int(roi[0] + roi[2])]
-            color_mean = np.mean(roi_hsv, axis=(0, 1))
-            color_mean = np.uint8(color_mean)
-
-            greenLower = np.array([max(color_mean[0] - 10, 0), 50, 50])
-            greenUpper = np.array([min(color_mean[0] + 10, 179), 255, 255])
-            print(f"Selected color: {color_mean}, Lower bound: {greenLower}, Upper bound: {greenUpper}")
-            color_selected = True
-            cv2.destroyAllWindows()
-        else:
-            print("No ROI selected. Exiting...")
-            cv2.destroyAllWindows()
-            return False
-    return True
-
-def track_color(frame, greenLower, greenUpper):
-    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
-    mask = cv2.inRange(hsv, greenLower, greenUpper)
-
-    kernel = np.ones((5, 5), np.uint8)
-    mask = cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernel)
-    mask = cv2.morphologyEx(mask, cv2.MORPH_CLOSE, kernel)
-
-    contours, _ = cv2.findContours(mask.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-
-    for contour in contours:
-        if cv2.contourArea(contour) > 200:
-            x, y, w, h = cv2.boundingRect(contour)
-            cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)
-
-    return frame, mask
-
-def main():
-    init_camera()
     color_selected = False
+    color_lower = None
+    color_upper = None
 
     while True:
+        # 从摄像头读取一帧
         ret, frame = cap.read()
         if not ret:
             break
 
+        # 如果还没有选择颜色区域，显示当前帧并允许用户选择
         if not color_selected:
-            if not select_color_area(frame):
+            cv2.imshow('Select color area', frame)
-                break
+
+            # 用户按下'c'键时，截取当前帧并允许用户选择ROI
+            if cv2.waitKey(1) & 0xFF == ord('c'):
+                # 转换到HSV
+                hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
+                roi = cv2.selectROI('Select color area', hsv)
+                if roi != (0, 0, 0, 0):
+                    # 计算ROI的平均颜色
+                    roi_hsv = hsv[int(roi[1]):int(roi[1] + roi[3]), int(roi[0]):int(roi[0] + roi[2])]
+                    color_mean = np.mean(roi_hsv, axis=(0, 1))
+                    color_mean = np.uint8(color_mean)
+
+                    # 创建颜色范围
+                    # color_range = 10
+                    color_lower = np.array(
+                        [max(color_mean[0] - 30, 0), max(color_mean[1] - 60, 0), max(color_mean[2] - 55, 0)])
+                    color_upper = np.array(
+                        [min(color_mean[0] + 10, 179), 255, 255])
+                    # print(f"Selected color: {color_mean}, Lower bound: {greenLower}, Upper bound: {greenUpper}")
+                    color_selected = True
+                    cv2.destroyWindow('Select color area')
+                else:
+                    print("No ROI selected. Exiting...")
+                    break
+
         else:
-            frame, mask = track_color(frame, greenLower, greenUpper)
+            # 颜色区域已选择，进行颜色追踪
+            # 将BGR图像转换为HSV
+            hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
+
+            # 创建掩码
+            mask = cv2.inRange(hsv, color_lower, color_upper)
+
+            # 应用开运算和闭运算去除噪声
+            kernel = np.ones((5, 5), np.uint8)
+            mask = cv2.morphologyEx(mask, cv2.MORPH_OPEN, kernel)
+            mask = cv2.morphologyEx(mask, cv2.MORPH_CLOSE, kernel)
+
+            # 查找轮廓
+            contours, _ = cv2.findContours(mask.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+
+            # 遍历轮廓，画出边界框
+            for contour in contours:
+                if cv2.contourArea(contour) > 300:
+                    x, y, w, h = cv2.boundingRect(contour)
+                    cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)
+            print(color_lower, color_upper)
+
+            # 显示原图和结果图像
             cv2.imshow('Original', frame)
             cv2.imshow('Mask', mask)
 
+            # 按'q'键退出循环
             if cv2.waitKey(1) & 0xFF == ord('q'):
                 break
 
+    # 释放资源并关闭所有窗口
     cap.release()
     cv2.destroyAllWindows()
 
-if __name__ == "__main__":
+if __name__ == '__main__':
     main()
+

工程/Color_tracking.py

@@ -6,12 +6,12 @@ from threading import Thread
 
 # 颜色的HSV阈值
 color_hsv_ranges = {
-    'blue': {'lower': np.array([100, 150, 50], dtype=np.uint8),
+    'blue': {'lower': np.array([80, 170, 192], dtype=np.uint8),
-             'upper': np.array([140, 255, 255], dtype=np.uint8)},
+             'upper': np.array([120, 255, 255], dtype=np.uint8)},
-    'green': {'lower': np.array([35, 46, 120], dtype=np.uint8),
+    'green': {'lower': np.array([25, 50, 147], dtype=np.uint8),
-              'upper': np.array([60, 255, 255], dtype=np.uint8)},
+              'upper': np.array([55, 255, 255], dtype=np.uint8)},
-    'red': {'lower': np.array([0, 116, 164], dtype=np.uint8),
+    'red': {'lower': np.array([0, 112, 194] , dtype=np.uint8),
-            'upper': np.array([180, 255, 255], dtype=np.uint8)},
+            'upper': np.array([13, 255, 255], dtype=np.uint8)},
     'custom': {'lower': np.array([0, 0, 0], dtype=np.uint8),
                'upper': np.array([180, 255, 255], dtype=np.uint8)}
 }
@@ -79,6 +79,7 @@ def exit_app():
     global running, cap
     running = False
     cap.release()
+    cv2.destroyAllWindows()
     root.quit()
 
 def initialize_gui():

工程/invoke.py

@@ -1,3 +1,34 @@
 import Color_tracking
-import PyQt6
+import Area_tracking
-Color_tracking.main()
+import sys
+from PyQt6.QtWidgets import QApplication, QWidget, QPushButton
+
+def quit_app():
+    sys.exit(app.quit())
+
+def main1():
+    print("main1 function called")
+
+def main2():
+    print("main2 function called")
+
+app = QApplication(sys.argv)
+
+widget = QWidget()
+widget.resize(300, 200)
+
+button1 = QPushButton('自选单颜色追踪', widget)
+button1.move(50, 50)
+button1.clicked.connect(Color_tracking.main)
+
+button2 = QPushButton('自选区域颜色追踪', widget)
+button2.move(50, 100)
+button2.clicked.connect(Area_tracking.main)
+
+quit_button = QPushButton('退出', widget)
+quit_button.move(150, 50)
+quit_button.clicked.connect(quit_app)
+
+widget.show()
+
+sys.exit(app.exec())