save-nparray-img

1
        # print("[jf]", obs.size())
2
        # [jf] torch.Size([4, 480, 480])
3
        def func():
4
            ...
5
            print("[jf] #20 save im")
6
            import matplotlib.pyplot as plt
7
            arr = obs[:3].cpu().numpy()
8
            arr = (arr * 255).astype(np.uint8)
9
            img = np.transpose(arr, (1, 2, 0))
10
            SAVE_PATH = '/home/junjie/repos/cloth-funnels/dist'
11
            plt.imsave(os.path.join(SAVE_PATH, f'{time.strftime("%Y-%m-%d %H-%M-%S")+" "+str(time.time())}.png'), img)
12

13
        # func()