65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
|
'''
|
|||
|
|
Program: MT_imgWatermark.py (Report comments/bugs to chikh@yuntech.edu.tw)
|
|||
|
|
Function:
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
import cv2 #引入OpenCV影像處理的相關套件
|
|||
|
|
from datetime import datetime #引入讀取機器時間相關套件
|
|||
|
|
import threading #引入多執行緒相關套件
|
|||
|
|
import time #for time.sleep()
|
|||
|
|
|
|||
|
|
threadList, imgList = [], []
|
|||
|
|
sem = threading.Semaphore(1)
|
|||
|
|
|
|||
|
|
#----- 設定要添加的文字：開始 -----#
|
|||
|
|
position = (10,40) #文字左下角的座標
|
|||
|
|
font = cv2.FONT_HERSHEY_COMPLEX
|
|||
|
|
fontScale = 0.5 #指定文字大小的比例
|
|||
|
|
color = (0,255,0) #以RGB三原色設定綠色文字；RGB色表可見 https://bit.ly/2ymSOJF
|
|||
|
|
thickness = 1 #文字的線寬
|
|||
|
|
#----- 設定要添加的文字：結束 -----#
|
|||
|
|
|
|||
|
|
def camCapture(ID):
|
|||
|
|
global imgList
|
|||
|
|
cap = cv2.VideoCapture(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)
|
|||
|
|
sem.acquire()
|
|||
|
|
imgList.append((ID,datetime.now().strftime("%H-%M-%S"),frame))
|
|||
|
|
cv2.imwrite('./opencvData/[%d] img-%s.jpg'%(imgList[-1][0],imgList[-1][1]),frame) #影像輸出為檔案儲存
|
|||
|
|
sem.release()
|
|||
|
|
cap.release() #釋放攝像頭設備
|
|||
|
|
|
|||
|
|
def procImage():
|
|||
|
|
global imgList
|
|||
|
|
while True:
|
|||
|
|
sem.acquire()
|
|||
|
|
if len(imgList) > 0:
|
|||
|
|
img = imgList[0][2]
|
|||
|
|
text = "Cam%d-%s"%(imgList[0][0],imgList[0][1])
|
|||
|
|
cv2.putText(img,text,position,font,fontScale,color,thickness) #在影像上添加文字
|
|||
|
|
cv2.imshow('Camera%d'%imgList[0][0],img) #顯示影像
|
|||
|
|
cv2.imwrite('./opencvData/Cam%d-%s.jpg'%(imgList[0][0],imgList[0][1]),img)
|
|||
|
|
imgList.pop(0)
|
|||
|
|
sem.release()
|
|||
|
|
if cv2.waitKey(1) & 0xFF in (27,81,113): break
|
|||
|
|
time.sleep(0.1)
|
|||
|
|
|
|||
|
|
for i in range(2):
|
|||
|
|
threadList.append(threading.Thread(target=camCapture,args=(i,)))
|
|||
|
|
threadList[i].start()
|
|||
|
|
|
|||
|
|
threadList.append(threading.Thread(target=procImage))
|
|||
|
|
threadList[-1].start()
|
|||
|
|
|
|||
|
|
for i in range(3):
|
|||
|
|
threadList[i].join() #wait for the thread specified by threadList[i] to terminate
|
|||
|
|
|
|||
|
|
cv2.destroyAllWindows() #關閉所有視窗
|