使用OpenCV在按下Enter鍵時截圖并保存到指定文件夾

在這篇博客中，我們將介紹如何使用OpenCV庫來實現一個簡單的功能：在按下Enter鍵時從攝像頭截圖并保存到指定的文件夾中。這個功能可以用于各種應用，例如監控系統、視頻捕捉等。

前置條件

在開始之前，請確保你已經安裝了OpenCV庫。如果沒有安裝，可以使用以下命令進行安裝：

pip install opencv-python

代碼實現

以下是完整的代碼實現：

import cv2
import os# 創建保存截圖的文件夾
if not os.path.exists('image'):os.makedirs('image')# 打開攝像頭
cap = cv2.VideoCapture(0)count = 0while True:# 讀取幀ret, frame = cap.read()# 顯示幀cv2.imshow('Camera', frame)# 按下'Enter'鍵截圖并保存if cv2.waitKey(1) & 0xFF == 13:  # 13 是Enter鍵的ASCII碼count += 1filename = f'image/screenshot_{count}.png'cv2.imwrite(filename, frame)print(f'Screenshot saved as {filename}')# 按下'q'鍵退出if cv2.waitKey(1) & 0xFF == ord('q'):break# 釋放攝像頭并關閉窗口
cap.release()
cv2.destroyAllWindows()

代碼解析

1. 導入庫：首先，我們需要導入OpenCV庫和os庫。

   import cv2
   import os
   

2. 創建保存截圖的文件夾：檢查是否存在名為image的文件夾，如果不存在則創建。

   if not os.path.exists('image'):os.makedirs('image')
   

3. 打開攝像頭：使用cv2.VideoCapture(0)打開默認攝像頭。

   cap = cv2.VideoCapture(0)
   

4. 讀取和顯示幀：在一個循環中不斷讀取攝像頭的幀并顯示。

   while True:ret, frame = cap.read()cv2.imshow('Camera', frame)
   

5. 按下Enter鍵截圖并保存：檢查是否按下了Enter鍵（ASCII碼為13），如果按下則保存當前幀到image文件夾中。

   if cv2.waitKey(1) & 0xFF == 13:count += 1filename = f'image/screenshot_{count}.png'cv2.imwrite(filename, frame)print(f'Screenshot saved as {filename}')
   

6. 按下’q’鍵退出：檢查是否按下了’q’鍵，如果按下則退出循環。

   if cv2.waitKey(1) & 0xFF == ord('q'):break
   

7. 釋放資源：釋放攝像頭并關閉所有OpenCV窗口。

   cap.release()
   cv2.destroyAllWindows()
   

結論

通過以上步驟，我們實現了一個簡單的功能，可以在按下Enter鍵時從攝像頭截圖并保存到指定文件夾中。這個功能可以根據需要進行擴展，例如添加更多的鍵盤控制、保存不同格式的圖片等。希望這篇博客對你有所幫助！

---

希望這篇博客對你有幫助！如果你有任何問題或需要進一步的幫助，請隨時告訴我。