fix: code format error

This commit is contained in:
mjq2020 2025-04-07 05:03:44 +01:00
parent 814f2a2935
commit 9d7755dc7f

View File

@ -458,17 +458,14 @@ def yuv_crop_and_resize(frame, region, height=None):
# copy u2 # copy u2
yuv_cropped_frame[ yuv_cropped_frame[
size + uv_channel_y_offset : size + uv_channel_y_offset + uv_crop_height, size + uv_channel_y_offset : size + uv_channel_y_offset + uv_crop_height,
size // 2 size // 2 + uv_channel_x_offset : size // 2
+ uv_channel_x_offset : size // 2
+ uv_channel_x_offset + uv_channel_x_offset
+ uv_crop_width, + uv_crop_width,
] = frame[u2[1] : u2[3], u2[0] : u2[2]] ] = frame[u2[1] : u2[3], u2[0] : u2[2]]
# copy v1 # copy v1
yuv_cropped_frame[ yuv_cropped_frame[
size size + size // 4 + uv_channel_y_offset : size
+ size // 4
+ uv_channel_y_offset : size
+ size // 4 + size // 4
+ uv_channel_y_offset + uv_channel_y_offset
+ uv_crop_height, + uv_crop_height,
@ -477,14 +474,11 @@ def yuv_crop_and_resize(frame, region, height=None):
# copy v2 # copy v2
yuv_cropped_frame[ yuv_cropped_frame[
size size + size // 4 + uv_channel_y_offset : size
+ size // 4
+ uv_channel_y_offset : size
+ size // 4 + size // 4
+ uv_channel_y_offset + uv_channel_y_offset
+ uv_crop_height, + uv_crop_height,
size // 2 size // 2 + uv_channel_x_offset : size // 2
+ uv_channel_x_offset : size // 2
+ uv_channel_x_offset + uv_channel_x_offset
+ uv_crop_width, + uv_crop_width,
] = frame[v2[1] : v2[3], v2[0] : v2[2]] ] = frame[v2[1] : v2[3], v2[0] : v2[2]]
@ -1043,7 +1037,6 @@ class SharedMemoryResultManager(FrameManager):
size = int.from_bytes(shm.buf[0:4], byteorder="little") size = int.from_bytes(shm.buf[0:4], byteorder="little")
if size <= 0: if size <= 0:
return None return None
batch_num = size // self.result_size // 4 batch_num = size // self.result_size // 4