async def recv(self):
frame = await self.received.get()
self.counter += 1
if (self.counter % 100) > 50:
# apply image processing to frame
if self.transform == 'edges':
img = frame_to_bgr(frame)
edges = cv2.Canny(img, 100, 200)
return frame_from_gray(edges)
elif self.transform == 'rotate':
img = frame_to_bgr(frame)
rows, cols, _ = img.shape
M = cv2.getRotationMatrix2D((cols / 2, rows / 2), self.counter * 7.2, 1)
rotated = cv2.warpAffine(img, M, (cols, rows))
return frame_from_bgr(rotated)
elif self.transform == 'green':
return VideoFrame(width=frame.width, height=frame.height)
else:
return frame
else:
# return raw frame
return frame
def test_frame_from_gray(self):
image = numpy.full((480, 640), 0, numpy.uint8)
frame = frame_from_gray(image)
self.assertEqual(len(frame.data), 460800)
self.assertEqual(frame.width, 640)
self.assertEqual(frame.height, 480)