pyorbbec 1.0.1.7__py3-none-any.whl → 1.0.1.8__py3-none-any.whl

pyorbbecsdk/examples/utils.py

@@ -0,0 +1,127 @@
+# ******************************************************************************
+#  Copyright (c) 2024 Orbbec 3D Technology, Inc
+#  
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.  
+#  You may obtain a copy of the License at
+#  
+#      http:# www.apache.org/licenses/LICENSE-2.0
+#  
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#  See the License for the specific language governing permissions and
+#  limitations under the License.
+# ******************************************************************************
+from typing import Union, Any, Optional
+
+import cv2
+import numpy as np
+
+from pyorbbecsdk import FormatConvertFilter, VideoFrame
+from pyorbbecsdk import OBFormat, OBConvertFormat
+
+
+def yuyv_to_bgr(frame: np.ndarray, width: int, height: int) -> np.ndarray:
+    yuyv = frame.reshape((height, width, 2))
+    bgr_image = cv2.cvtColor(yuyv, cv2.COLOR_YUV2BGR_YUY2)
+    return bgr_image
+
+
+def uyvy_to_bgr(frame: np.ndarray, width: int, height: int) -> np.ndarray:
+    uyvy = frame.reshape((height, width, 2))
+    bgr_image = cv2.cvtColor(uyvy, cv2.COLOR_YUV2BGR_UYVY)
+    return bgr_image
+
+
+def i420_to_bgr(frame: np.ndarray, width: int, height: int) -> np.ndarray:
+    y = frame[0:height, :]
+    u = frame[height:height + height // 4].reshape(height // 2, width // 2)
+    v = frame[height + height // 4:].reshape(height // 2, width // 2)
+    yuv_image = cv2.merge([y, u, v])
+    bgr_image = cv2.cvtColor(yuv_image, cv2.COLOR_YUV2BGR_I420)
+    return bgr_image
+
+
+def nv21_to_bgr(frame: np.ndarray, width: int, height: int) -> np.ndarray:
+    y = frame[0:height, :]
+    uv = frame[height:height + height // 2].reshape(height // 2, width)
+    yuv_image = cv2.merge([y, uv])
+    bgr_image = cv2.cvtColor(yuv_image, cv2.COLOR_YUV2BGR_NV21)
+    return bgr_image
+
+
+def nv12_to_bgr(frame: np.ndarray, width: int, height: int) -> np.ndarray:
+    y = frame[0:height, :]
+    uv = frame[height:height + height // 2].reshape(height // 2, width)
+    yuv_image = cv2.merge([y, uv])
+    bgr_image = cv2.cvtColor(yuv_image, cv2.COLOR_YUV2BGR_NV12)
+    return bgr_image
+
+
+def determine_convert_format(frame: VideoFrame):
+    if frame.get_format() == OBFormat.I420:
+        return OBConvertFormat.I420_TO_RGB888
+    elif frame.get_format() == OBFormat.MJPG:
+        return OBConvertFormat.MJPG_TO_RGB888
+    elif frame.get_format() == OBFormat.YUYV:
+        return OBConvertFormat.YUYV_TO_RGB888
+    elif frame.get_format() == OBFormat.NV21:
+        return OBConvertFormat.NV21_TO_RGB888
+    elif frame.get_format() == OBFormat.NV12:
+        return OBConvertFormat.NV12_TO_RGB888
+    elif frame.get_format() == OBFormat.UYVY:
+        return OBConvertFormat.UYVY_TO_RGB888
+    else:
+        return None
+
+
+def frame_to_rgb_frame(frame: VideoFrame) -> Union[Optional[VideoFrame], Any]:
+    if frame.get_format() == OBFormat.RGB:
+        return frame
+    convert_format = determine_convert_format(frame)
+    if convert_format is None:
+        print("Unsupported format")
+        return None
+    print("covert format: {}".format(convert_format))
+    convert_filter = FormatConvertFilter()
+    convert_filter.set_format_convert_format(convert_format)
+    rgb_frame = convert_filter.process(frame)
+    if rgb_frame is None:
+        print("Convert {} to RGB failed".format(frame.get_format()))
+    return rgb_frame
+
+
+def frame_to_bgr_image(frame: VideoFrame) -> Union[Optional[np.array], Any]:
+    width = frame.get_width()
+    height = frame.get_height()
+    color_format = frame.get_format()
+    data = np.asanyarray(frame.get_data())
+    image = np.zeros((height, width, 3), dtype=np.uint8)
+    if color_format == OBFormat.RGB:
+        image = np.resize(data, (height, width, 3))
+        image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+    elif color_format == OBFormat.BGR:
+        image = np.resize(data, (height, width, 3))
+        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+    elif color_format == OBFormat.YUYV:
+        image = np.resize(data, (height, width, 2))
+        image = cv2.cvtColor(image, cv2.COLOR_YUV2BGR_YUYV)
+    elif color_format == OBFormat.MJPG:
+        image = cv2.imdecode(data, cv2.IMREAD_COLOR)
+    elif color_format == OBFormat.I420:
+        image = i420_to_bgr(data, width, height)
+        return image
+    elif color_format == OBFormat.NV12:
+        image = nv12_to_bgr(data, width, height)
+        return image
+    elif color_format == OBFormat.NV21:
+        image = nv21_to_bgr(data, width, height)
+        return image
+    elif color_format == OBFormat.UYVY:
+        image = np.resize(data, (height, width, 2))
+        image = cv2.cvtColor(image, cv2.COLOR_YUV2BGR_UYVY)
+    else:
+        print("Unsupported color format: {}".format(color_format))
+        return None
+    return image

pyorbbec-1.0.1.7.dist-info/RECORD

@@ -1,18 +0,0 @@
-pyorbbec-1.0.1.7.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-pyorbbec-1.0.1.7.dist-info/licenses/NOTICE,sha256=HGutM36evF5jV29JFjgiEnEDjEzOJ2D7rcTKa-BRlhg,34
-pyorbbecsdk/OrbbecSDK.dll,sha256=XLrZ9iNk4LBeeTfF-Vg3re7d5QpMX8zm_MItJoSwDHI,5578752
-pyorbbecsdk/OrbbecSDK.lib,sha256=-YPC20qqE2pHA4O9LD5r1PnRDODKC3fsqJi1D_g-Zdc,87364
-pyorbbecsdk/OrbbecSDKConfig.xml,sha256=K8UO1coFe_DK_ZXFdjkyoeq0pL4BHYInhD_yv4BiSt4,116521
-pyorbbecsdk/__init__.py,sha256=03B3LOLwYnfJD60j-wmz1nwuOMHLPnO3fJtJN8-tfEM,558
-pyorbbecsdk/__version__.py,sha256=A7lTxKw06iHB99ygtvk_bxIr7wFSg3VghDAYWNEB_Sk,324
-pyorbbecsdk/pyorbbecsdk.cp313-win_amd64.pyd,sha256=kFT29d743T-iy1QHWmvElOvkfA3-86RcM9GK3JrrwPc,1330688
-pyorbbecsdk/extensions/depthengine/depthengine.dll,sha256=wBTUpKUOdczTdfocLQP9HIyGu-erP24fgQLK8QPkwww,355328
-pyorbbecsdk/extensions/depthengine/depthengine.lib,sha256=jGOeCkxG97Dxe3DXWdx9NjMO8pDhWiVo3uJpO5x_b7A,5194
-pyorbbecsdk/extensions/filters/FilterProcessor.dll,sha256=rq4cw-PbOAKwzGH_LsngxBHnpOCC9MNHojbQGDZeO64,93696
-pyorbbecsdk/extensions/filters/ob_priv_filter.dll,sha256=qSGeHXBpq4lKf-dwbbJPsSErE-IBiAov3GRPDszfS7c,69120
-pyorbbecsdk/extensions/firmwareupdater/firmwareupdater.dll,sha256=MBblwWX3ObOmQCynki4MUOvHPNet7Je9Qo72doHmpew,192512
-pyorbbecsdk/extensions/frameprocessor/ob_frame_processor.dll,sha256=LseGWdK6xzq9Ab4QClFqVAz0SsTeDGTgg3hImJRCiA8,149504
-pyorbbec-1.0.1.7.dist-info/METADATA,sha256=wUqqaEjnNJhTqBdYZy43v_NCuDp465Lx_-6UEHJYJ8k,631
-pyorbbec-1.0.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pyorbbec-1.0.1.7.dist-info/top_level.txt,sha256=VtEe2PLpfyn__4hM0LpuzMtTVwgzui9ou5Tox-0-nDI,12
-pyorbbec-1.0.1.7.dist-info/RECORD,,