Skip to content

Instantly share code, notes, and snippets.

@radames
Last active April 17, 2023 15:34
Show Gist options
  • Save radames/1e7c794842755683162b to your computer and use it in GitHub Desktop.
Save radames/1e7c794842755683162b to your computer and use it in GitHub Desktop.
OpenCV VideoCapture running on PyGame - repo ref https://github.com/radames/opencv_video_to_pygame
from pygame.locals import KEYDOWN, K_ESCAPE, K_q
import pygame
import cv2
import sys
camera = cv2.VideoCapture(1)
pygame.init()
pygame.display.set_caption("OpenCV camera stream on Pygame")
screen = pygame.display.set_mode([1280, 720])
try:
while True:
ret, frame = camera.read()
screen.fill([0, 0, 0])
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
frame = frame.swapaxes(0, 1)
pygame.surfarray.blit_array(screen, frame)
pygame.display.update()
for event in pygame.event.get():
if event.type == pygame.QUIT:
sys.exit(0)
elif event.type == KEYDOWN:
if event.key == K_ESCAPE or event.key == K_q:
sys.exit(0)
except (KeyboardInterrupt, SystemExit):
pygame.quit()
cv2.destroyAllWindows()
Copy link

ghost commented Oct 10, 2020

I am getting this error:
"ValueError: array must match surface dimensions"
What should I do?

@WiktorLigeza
Copy link

I am getting this error:
"ValueError: array must match surface dimensions"
What should I do?

Change screen size to your camera output size screen = pygame.display.set_mode([640, 480]) worked for me.
You can get size of frame easily E.g: print(frame.shape)

@mathcat4
Copy link

I can't thank you enough for this code. Keep it up.

@angelman7
Copy link

angelman7 commented Feb 6, 2021

Guys if anyone had this error

cv2.error: OpenCV(4.5.1) C:\Users\appveyor\AppData\Local\Temp\1\pip-req-build-kh7iq4w7\opencv\modules\imgproc\src\color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cv::cvtColor'

add frame = cv2.VideoCapture(0) below ret, frame = camera.read()

@radames
Copy link
Author

radames commented Feb 9, 2021

hi @angelman7 can't help much but it could be related to your camera image streaming,
read more https://stackoverflow.com/questions/52676020/opencv-src-empty-in-function-cvtcolor-error

@mathcat4
Copy link

Can you show it the other way too?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment