Getting instance and class segments from GTA V Gamehook

gta_gamehook_segments.py

import numpy as np
import matplotlib.pyplot as plt

filename = 'C:\\Users\\Benjamin Tan\\Desktop\\00016351_object_id.raw'

vals = np.linspace(0,1,256)
np.random.shuffle(vals)
cmap = plt.cm.colors.ListedColormap(plt.cm.jet(vals))

# Load the object_id.raw file and skip the first 4 bytes
A = np.fromfile(filename, dtype='int32', sep='')[4:]
A = A.reshape([600, 800])
# shifting the bits to differentiate the segments
B = A & ((1<<28)-1)

# This shows instance segments
plt.imshow(B, cmap=cmap); plt.show()

# shifting the bits to get the class labels
C = ((A >> 28) & 0xf)
# >>> print(set(x for x in B.reshape(-1)))
# {0, 1, 2, 3} <- class labels
plt.imshow(C, cmap=cmap); plt.show()