Data¶

class pystk.RenderData¶

SuperTuxKart rendering output

property depth → numpy.ndarray¶: Depth image of the kart (memoryview[float] screen_height x screen_width)

property image → numpy.ndarray¶: Color image of the kart (memoryview[uint8] screen_height x screen_width x 3)

property instance → numpy.ndarray¶: Instance labels (memoryview[uint32] screen_height x screen_width)

Each instance label is spit into an ObjectType and instance label. Right shift (>>) the instance label by ObjectType.object_type_shift to retrieve the object type.

class pystk.ObjectType¶

object_type_shift¶: Number of bits for the instance label (shift of the object type)

N = 10¶

background = 3¶

bomb = 6¶

kart = 1¶

property name → str¶

nitro = 5¶

object = 7¶

pickup = 4¶

projectile = 8¶

track = 2¶

unknown = 9¶

property value → int¶