michael/scrypted
1
0
Fork 0
mirror of https://github.com/koush/scrypted.git synced 2026-05-06 06:00:29 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

opencv: fixup color conversion fast path

Browse Source
This commit is contained in:
Koushik Dutta
2023-02-11 08:52:11 -08:00
parent 8142893395
commit efdeb95bbe
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
plugins/opencv/src/opencv/__init__.py
View File

@@ -213,12 +213,10 @@ class OpenCVPlugin(DetectPlugin):
raise Exception('can not run motion detection on image')
def run_detection_avframe(self, detection_session: DetectionSession, avframe, settings: Any, src_size, convert_to_src_size) -> Tuple[ObjectsDetected, Any]:
format = avframe.format
if format != 'yuv420p' or format != 'yuvj420p':
format = 'yuvj420p'
if avframe.format.name != 'yuv420p' and avframe.format.name != 'yuvj420p':
mat = avframe.to_ndarray(format='gray8')
else:
format = None
mat = avframe.to_ndarray(format=format)
mat = np.ndarray((avframe.height, avframe.width, self.pixelFormatChannelCount), buffer=avframe.planes[0], dtype=np.uint8)
detections = self.detect(
detection_session, mat, settings, src_size, convert_to_src_size)
if not detections or not len(detections['detections']):