How to plot the data

[JayGao1219]

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

[JayGao1219]

Here comes my result which is very strange