mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-04-07 10:24:02 +08:00
793f8fee32
getRamImage() returns panda3d's internal BGRA format. on macOS this
produces swapped red/blue channels in the sim camera feed.
getRamImageAs("RGBA") requests explicit RGBA reordering from panda3d,
correct on all platforms. no-op where internal format is already RGBA.
ref: https://docs.panda3d.org/1.10/python/reference/panda3d.core.Texture#panda3d.core.Texture.getRamImageAs
fixes #37526
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
import numpy as np
|
|
|
|
from metadrive.component.sensors.rgb_camera import RGBCamera
|
|
from panda3d.core import Texture, GraphicsOutput
|
|
|
|
|
|
class CopyRamRGBCamera(RGBCamera):
|
|
"""Camera which copies its content into RAM during the render process, for faster image grabbing."""
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.cpu_texture = Texture()
|
|
self.buffer.addRenderTexture(self.cpu_texture, GraphicsOutput.RTMCopyRam)
|
|
|
|
def get_rgb_array_cpu(self):
|
|
origin_img = self.cpu_texture
|
|
img = np.frombuffer(origin_img.getRamImageAs("RGBA").getData(), dtype=np.uint8)
|
|
img = img.reshape((origin_img.getYSize(), origin_img.getXSize(), 4))
|
|
img = img[:, :, :3]
|
|
# img = np.swapaxes(img, 1, 0)
|
|
img = img[::-1] # Flip on vertical axis
|
|
return img
|
|
|
|
|
|
class RGBCameraWide(CopyRamRGBCamera):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
lens = self.get_lens()
|
|
lens.setFov(120)
|
|
lens.setNear(0.1)
|
|
|
|
|
|
class RGBCameraRoad(CopyRamRGBCamera):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
lens = self.get_lens()
|
|
lens.setFov(40)
|
|
lens.setNear(0.1)
|