49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
|
|
||
|
|
quit()
|
||
|
|
#take one image out
|
||
|
|
heatmap = np.squeeze(heatmap[0])
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
import matplotlib.pyplot as plt
|
||
|
|
|
||
|
|
# Fixing random state for reproducibility
|
||
|
|
np.random.seed(19680801)
|
||
|
|
|
||
|
|
|
||
|
|
class IndexTracker:
|
||
|
|
def __init__(self, ax, X):
|
||
|
|
self.ax = ax
|
||
|
|
ax.set_title('use scroll wheel to navigate images')
|
||
|
|
|
||
|
|
self.X = X
|
||
|
|
rows, cols, self.slices = X.shape
|
||
|
|
self.ind = self.slices//2
|
||
|
|
|
||
|
|
self.im = ax.imshow(self.X[:, :, self.ind], cmap='jet')
|
||
|
|
self.update()
|
||
|
|
|
||
|
|
def on_scroll(self, event):
|
||
|
|
print("%s %s" % (event.button, event.step))
|
||
|
|
if event.button == 'up':
|
||
|
|
self.ind = (self.ind + 1) % self.slices
|
||
|
|
else:
|
||
|
|
self.ind = (self.ind - 1) % self.slices
|
||
|
|
self.update()
|
||
|
|
|
||
|
|
def update(self):
|
||
|
|
self.im.set_data(self.X[:, :, self.ind])
|
||
|
|
self.ax.set_ylabel('slice %s' % self.ind)
|
||
|
|
self.im.axes.figure.canvas.draw()
|
||
|
|
|
||
|
|
plt.figure(0)
|
||
|
|
fig, ax = plt.subplots(1, 1)
|
||
|
|
tracker = IndexTracker(ax, heatmap[:,:,:,5])
|
||
|
|
fig.canvas.mpl_connect('scroll_event', tracker.on_scroll)
|
||
|
|
plt.show()
|
||
|
|
|
||
|
|
plt.figure(1)
|
||
|
|
fig, ax = plt.subplots(1, 1)
|
||
|
|
tracker = IndexTracker(ax, heatmap[:,:,:,3])
|
||
|
|
fig.canvas.mpl_connect('scroll_event', tracker.on_scroll)
|
||
|
|
plt.show()
|