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

Fixing index out of range

Browse Source
This commit is contained in:
Endre Simo
2019-02-05 06:52:56 +02:00
parent e211d446ef
commit 4d56440cb9
1 changed files with 7 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
examples/python/demo.py
View File

@@ -52,17 +52,15 @@ while(True):
ret, frame = cap.read()
pixs = numpy.ascontiguousarray(frame[:, :, 1].reshape((frame.shape[0], frame.shape[1])))
# Check if camera is intialized by checking if pixel array is not empty.
if not numpy.any(pixs):
continue
dets = process_frame(pixs) # pixs needs to be numpy.uint8 array
if dets is not None:
for det in dets:
cv2.circle(frame, (int(det[1]), int(det[0])), int(det[2]/2.0), (0, 0, 255), 2)
# Verify if camera is intialized by checking if pixel array is not empty.
if numpy.any(pixs):
dets = process_frame(pixs) # pixs needs to be numpy.uint8 array
if dets is not None:
for det in dets:
cv2.circle(frame, (int(det[1]), int(det[0])), int(det[2]/2.0), (0, 0, 255), 2)
cv2.imshow('', frame)
if cv2.waitKey(5) & 0xFF == ord('q'):
break