Ошибка при рисовании анимации морской карты интенсивности для 3D-объема

Пытаясь визуализировать взаимную корреляцию между двумя томами, img_3D и mask_3D, используя Тепловая карта Seaborn и анимация из Matplotlib, чтобы визуализировать результат кросс-корреляции 3D как прогрессивную анимацию 2D-изображений, но я столкнулся с ошибкой. Подскажите, пожалуйста, как чтобы избавиться от этой ошибки и правильно визуализировать тепловые карты?

Заранее спасибо.

Traceback (most recent call last):
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\tkinter\__init__.py", line 1705, in __call__
    return self.func(*args)
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\site-packages\matplotlib\backends\_backend_tk.py", line 259, in resize
    self.draw()
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\site-packages\matplotlib\backends\backend_tkagg.py", line 9, in draw
    super(FigureCanvasTkAgg, self).draw()
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\site-packages\matplotlib\backends\backend_agg.py", line 392, in draw
    else nullcontext()):
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\contextlib.py", line 112, in __enter__
    return next(self.gen)
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\site-packages\matplotlib\backend_bases.py", line 2788, in _wait_cursor_for_draw_cm
    self.set_cursor(self._lastCursor)
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\site-packages\matplotlib\backends\_backend_tk.py", line 544, in set_cursor
    window.configure(cursor=cursord[cursor])
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\tkinter\__init__.py", line 1485, in configure
    return self._configure('configure', cnf, kw)
  File "C:\Users\User\AppData\Local\Programs\Python\Python37\lib\tkinter\__init__.py", line 1476, in _configure
    self.tk.call(_flatten((self._w, cmd)) + self._options(cnf))
_tkinter.TclError: invalid command name "."

Используемый код:

# Import Libraries
#====================================
import numpy as np
np.random.seed(0)
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
import nibabel as nib
from scipy.signal import correlate

import seaborn as sns
sns.set()
#===================================

img = np.load('img.npy')
act = np.load('act.npy')

# Mode : 'full', 'valid', 'same'

result = correlate(img, act,mode='same')

print(img.shape, act.shape, result.shape)

def updatefig(sl):
    for sl in range(result.shape[2]):
        print(sl,' / ',result.shape[2])
        sns.heatmap(result[...,sl],cbar=False)
        ax.set_title("frame {}".format(sl))
        # Note that using time.sleep does *not* work here!
        plt.pause(0.1)
fig, ax = plt.subplots()

ani = FuncAnimation(fig, updatefig, frames=range(result.shape[2]), interval=5, blit=True)

plt.show()

python matplotlib seaborn heatmap cross-correlation
person Bilal    schedule 15.06.2020    source источник

Ответы (1)


arrow_upward
2
arrow_downward

Проверьте этот код:

import numpy as np
np.random.seed(0)
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
from scipy.signal import correlate
import seaborn as sns
sns.set()

img = np.load('img.npy')
act = np.load('act.npy')

result = correlate(img, act, mode = 'same')

def updatefig(sl):
    ax.cla()
    print(sl + 1, ' / ', result.shape[2])
    sns.heatmap(result[..., sl], cbar = False)
    ax.set_title("frame {}".format(sl + 1))
    ax.axis('off')

fig, ax = plt.subplots()
ani = FuncAnimation(fig, updatefig, frames = result.shape[2], interval = 5)

plt.show()

что дает мне эту анимацию (я уменьшил вдвое анимацию, о которой говорится ниже, чтобы уменьшить размер файла до 2 МБ, приведенный выше код воспроизводит все 40 кадров):

введите здесь описание изображения

ИЗМЕНИТЬ

Чтобы добавить на тепловую карту фиксированную шкалу цветов, проверьте этот код:

import numpy as np
np.random.seed(0)
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
from scipy.signal import correlate
import seaborn as sns
sns.set()

img = np.load('img.npy')
act = np.load('act.npy')

result = correlate(img, act, mode = 'same')

def updatefig(sl):
    ax.cla()
    print(sl + 1, ' / ', result.shape[2])
    sns.heatmap(result[..., sl],
                ax = ax,
                cbar = True,
                cbar_ax = cbar_ax,
                vmin = result.min(),
                vmax = result.max())
    ax.set_title("frame {}".format(sl + 1))
    ax.axis('off')

grid_kws = {'width_ratios': (0.9, 0.05), 'wspace': 0.2}
fig, (ax, cbar_ax) = plt.subplots(1, 2, gridspec_kw = grid_kws, figsize = (10, 8))
ani = FuncAnimation(fig, updatefig, frames = result.shape[2], interval = 5)

plt.show()

который производит эту анимацию (вырезанную как предыдущую):

введите здесь описание изображения

person Andrea Blengino    schedule 15.06.2020
comment
Большое спасибо, я хотел бы спросить, как показать шкалу цветов для этой фигуры, пожалуйста? заранее спасибо. - person Bilal; 16.06.2020