How to plot the data

I have downloaded the data, and I try to plot the data  in this way.
```
U1_X=np.load('U1_features_X.npy')
data=U1_X[0][0]
plt.imshow(data, extent=[0,3,0,128], cmap='magma',vmin=np.min(data), vmax=np.max(data), aspect='auto')
```
But I did't get something like fig3 in your paper.
Can you tell me how to plot the data correctly? 
thanks