chris/frigate
1
0
Fork 0
forked from Github/frigate
Code Pull Requests Packages Activity

maintain aspect ratio for birdseye

Browse Source
This commit is contained in:
Blake Blackshear
2021-06-23 07:36:47 -05:00
parent 0ff037997f
commit fa61e9da29
3 changed files with 87 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
frigate/test/test_copy_yuv_to_position.py
View File

@@ -1,7 +1,7 @@
import cv2
import numpy as np
from unittest import TestCase, main
from frigate.util import copy_yuv_to_position
from frigate.util import get_yuv_crop, copy_yuv_to_position
class TestCopyYuvToPosition(TestCase):
@@ -11,14 +11,53 @@ class TestCopyYuvToPosition(TestCase):
self.source_yuv_frame = cv2.cvtColor(
self.source_frame_bgr, cv2.COLOR_BGR2YUV_I420
)
y, u1, u2, v1, v2 = get_yuv_crop(
self.source_yuv_frame.shape,
(
0,
0,
self.source_frame_bgr.shape[1],
self.source_frame_bgr.shape[0],
),
)
self.source_channel_dims = {
"y": y,
"u1": u1,
"u2": u2,
"v1": v1,
"v2": v2,
}
self.dest_frame_bgr = np.zeros((400, 800, 3), np.uint8)
self.dest_frame_bgr[:] = (112, 202, 50)
self.dest_frame_bgr[100:300, 200:600] = (255, 0, 0)
self.dest_yuv_frame = cv2.cvtColor(self.dest_frame_bgr, cv2.COLOR_BGR2YUV_I420)
def test_copy_yuv_to_position(self):
copy_yuv_to_position(1, self.dest_yuv_frame, 3)
def test_clear_position(self):
copy_yuv_to_position(self.dest_yuv_frame, (100, 100), (100, 100))
# cv2.imwrite(f"source_frame_yuv.jpg", self.source_yuv_frame)
# cv2.imwrite(f"dest_frame_yuv.jpg", self.dest_yuv_frame)
def test_copy_position(self):
copy_yuv_to_position(
self.dest_yuv_frame,
(100, 100),
(100, 200),
self.source_yuv_frame,
self.source_channel_dims,
)
# cv2.imwrite(f"source_frame_yuv.jpg", self.source_yuv_frame)
# cv2.imwrite(f"dest_frame_yuv.jpg", self.dest_yuv_frame)
def test_copy_position_full_screen(self):
copy_yuv_to_position(
self.dest_yuv_frame,
(0, 0),
(400, 800),
self.source_yuv_frame,
self.source_channel_dims,
)
# cv2.imwrite(f"source_frame_yuv.jpg", self.source_yuv_frame)
# cv2.imwrite(f"dest_frame_yuv.jpg", self.dest_yuv_frame)