关于实验中可视化的代码
import matplotlib.pyplot as plt
import cv2
f = pred_score.squeeze().cpu().detach().numpy()
f = f.transpose(2,1,0)[:,:,9]
a = np.stack([cv2.resize(f,(255,255),interpolation=cv2.INTER_CUBIC)])
a = np.squeeze(a)
plt.imshow(a)
plt.xticks([])
plt.yticks([])
plt.title([])
plt.axis('off')
plt.show()
m = pred_score.squeeze().cpu().detach().numpy()
sub_plot = [441, 442, 443, 444, 445, 446, 447, 448, 449, 4410]
for i in range(9):
f = m.transpose(2,1,0)[:,:,i]
a = np.stack([cv2.resize(f,(255,255),interpolation=cv2.INTER_CUBIC)])
a = np.squeeze(a)
plt.subplot(sub_plot[i])
plt.subplots_adjust(left=0.05, right=0.8, hspace=0.4)
plt.imshow(a)
plt.colorbar()
plt.xticks([])
plt.yticks([])
plt.title([])
plt.axis('off')
plt.show()