把存储类型为np.array类型的图片,动态的打印出来。
import os
import numpy as np
import matplotlib.pyplot as plt
# /home/kuka_data/data.npy中保存的是一个字典,共6560条数据
# 其中的obs是图像的矩阵,每张图片的大小是256x256x4
read_dictionary = np.load('/home/kuka_data/data.npy',allow_pickle=True).item()
a = len(read_dictionary['obs'])
print(np.array(read_dictionary["obs"]).shape)
# 注意 plt.ion() 不能省略
plt.ion()
for i in range(a):
# 遍历每张图片
img = read_dictionary['obs'][i]
plt.imshow(img)
plt.draw()
plt.show()
# 每停止0.1秒重新打印一次
plt.pause(0.1)
plt.clf()