package/pigo
1
0
Fork 0
mirror of https://github.com/esimov/pigo.git synced 2025-10-25 00:40:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: index out of range error on camera initialization on MacOS

Browse Source
This commit is contained in:
esimov
2022-11-10 11:10:50 +02:00
parent 0388e93ed9
commit 0c9a71c262
7 changed files with 43 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
examples/talk_detector/talkdet.py
View File

@@ -74,9 +74,10 @@ def process_frame(pixs):
return dets
# initialize the camera
width, height = 640, 480
cap = cv2.VideoCapture(0)
cap.set(cv2.CAP_PROP_FRAME_WIDTH, 640)
cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 480)
cap.set(cv2.CAP_PROP_FRAME_WIDTH, width)
cap.set(cv2.CAP_PROP_FRAME_HEIGHT, height)
showFaceDet = False
showPupil = True
@@ -91,8 +92,9 @@ while(True):
frame[:, :, 1].reshape((frame.shape[0], frame.shape[1])))
pixs = pixs.flatten()
# Verify if camera is intialized by checking if pixel array is not empty.
if np.any(pixs):
# We need to make sure that the whole frame size is transfered over Go,
# otherwise we might getting an index out of range panic error.
if len(pixs) == width*height:
dets = process_frame(pixs) # pixs needs to be numpy.uint8 array
if dets is not None:
@@ -130,7 +132,7 @@ while(True):
cv2.putText(frame, "Person talking...", (10, 30),
cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 0, 255), 1)
cv2.imshow('Talk detection demo', frame)
cv2.imshow('Talk detector', frame)
key = cv2.waitKey(1)
if key & 0xFF == ord('q'):