eda.py 文件源码

def plot_rgb_difference(self):
        """
        Plots difference between cover and steganographic images for each RGB
        colour channel.
        """
        f, axarr = plt.subplots(1, 3, figsize=(12, 4))
        for j, colour in enumerate(['Red', 'Green', 'Blue']):
            diff = self.I[:, :, j] - self.S[:, :, j]
            axarr[j].imshow(diff, cmap='{}s_r'.format(colour))
            axarr[j].set_title('{}'.format(colour))
            axarr[j].set_xticklabels([])
            axarr[j].set_yticklabels([])
        plt.show()