how do i visualize my synthesis pictures

ethnhe / raster_triangle

A simple renderer with z-buffer for synthesis data generating.

56 stars 22 forks source link

Open guess-who-ami opened 2 years ago

guess-who-ami commented 2 years ago

i just got 70000 .pkl files, how do i visualize these pictures ?

siatheindochinese commented 2 years ago

For each of the .pkl files, you can load them using python's pickle library. Just do the following (i.e for 0.pkl):

data = pkl.load(open("0.pkl","rb"))
depthmap = data["depth"]
rgbmap = data["rgb"]
mask = data["mask"]

Then just plot the 3 maps/mask using plt.imshow(....)