43 lines
1.9 KiB
Python
43 lines
1.9 KiB
Python
'''
|
||
Program: MT_imgAcquisition.py (Report comments/bugs to chikh@yuntech.edu.tw)
|
||
Function: 擴充imgAcquisitionSave.py成為多執行緒程式
|
||
Note: 本程式以筆電搭配另一個攝像鏡頭執行較能看出多執行緒的效果
|
||
'''
|
||
|
||
import cv2 #引入OpenCV影像處理的相關套件
|
||
from datetime import datetime #引入讀取機器時間相關套件
|
||
import threading #引入多執行緒相關套件
|
||
|
||
threadList = [] #儲存新建之多執行緒的串列
|
||
|
||
def camCapture(ID):
|
||
cap = cv2.VideoCapture(ID) #以ID選擇哪一個攝像鏡頭
|
||
i = 0
|
||
while True:
|
||
ret, frame = cap.read() #從攝像鏡頭讀取一幀影像
|
||
cv2.imshow("Window %d"%ID,frame) #重要!若未加入ID區分，二個鏡頭擷取的影像將重疊在同一視窗上
|
||
if cv2.waitKey(1) & 0xFF in (27,81,113): break #每隔1毫秒偵測是否有按鍵，若有，則讀取最末位元組內容，比對是否為<Esc>、'q'或'Q'，倘符合，則退出迴圈
|
||
i += 1
|
||
if (i%40 == 0):
|
||
i = 0 #重設i，避免持續增加i值造成溢位(overflow)
|
||
cv2.imwrite('./opencvData/[%d]-img-%s.jpg'%(ID,datetime.now().strftime("%H-%M-%S")),frame) #影像輸出為檔案儲存
|
||
cap.release() #釋放攝像鏡頭所佔資源
|
||
|
||
#------ 底下是主執行緒 ------#
|
||
howManyCams = int(input("\n這部機器有幾個攝像鏡頭？=> "))
|
||
if howManyCams == 0:
|
||
print("\n程式無法繼續運行，sorry.")
|
||
exit(0)
|
||
elif howManyCams == 1:
|
||
threadList.append(threading.Thread(target=camCapture,args=(0,)))
|
||
threadList[0].start() #啟動執行緒
|
||
threadList[0].join() #等候執行緒結束
|
||
else:
|
||
for i in range(2): #創建二個執行緒，每個執行緒處理單一鏡頭擷取的畫面
|
||
threadList.append(threading.Thread(target=camCapture,args=(i,)))
|
||
threadList[i].start() #啟動執行緒
|
||
for i in range(2):
|
||
threadList[i].join() #等候threadList[i]執行緒結束
|
||
|
||
cv2.destroyAllWindows() #關閉所有視窗
|