disp.py

import pickle; import numpy as np; import cv2
#im = pickle.load(open('data/multiagent_ae_kl__200000_10-12-2018_01-32/3/latent_2.pkl', 'rb'))
#im = pickle.load(open('data/pusher2_ae_kl__10000_10-12-2018_01-54/6/latent_0.pkl', 'rb'))
im = pickle.load(open('data/occlusion_ae_kl__250000_11-12-2018_04-26/9/sampled_3.pkl', 'rb'))
new_im = np.zeros((64, 64, 3), dtype=np.uint8)
num_active_layers = 0
for i in range(8):
    if np.max(im[:,:,i]) > 0:
        print(np.unique(im[:,:,i]))
        num_active_layers += 1
        color = [int(((i)%8)>=4),int(((i)%4)>=2),int(((i)%2)>=1)]
        new_im += np.round(im[:,:,i]).astype(np.uint8).reshape(64, 64, 1) \
            *255//2 \
            *np.array(color).astype(np.uint8).reshape(1, 1, 3)
print(num_active_layers)
cv2.imshow('im', new_im)
cv2.waitKey(0)
cv2.destroyAllWindows()