administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-02 17:25:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix the create_tensor_input cv2 resize

Browse Source
This commit is contained in:
YS 2022-04-05 14:47:36 +03:00
parent 846586d959
commit e037bcfcb4
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
frigate/video.py
View File

@ -79,9 +79,10 @@ def create_tensor_input(frame, model_shape, region):
cropped_frame = yuv_region_2_rgb(frame, region)
# Resize to the model_shape if needed
if cropped_frame.shape != (model_shape[0], model_shape[1], 3):
height, width = model_shape
if cropped_frame.shape != (height, width, 3):
cropped_frame = cv2.resize(
cropped_frame, dsize=model_shape, interpolation=cv2.INTER_LINEAR
cropped_frame, dsize=(width, height), interpolation=cv2.INTER_LINEAR
)
# Return a tensor of shape: [height, width, 3] in RGB format
return cropped_frame