Method get_diff

StegaPy/plugin/lsb/lsb_plugin.py:120–138

获取原始图像和隐写图像的差异

(self, stego_data: bytes, stego_filename: Optional[str],
                cover_data: bytes, cover_filename: Optional[str],
                diff_filename: Optional[str])

        return 0.0
    
    def get_diff(self, stego_data: bytes, stego_filename: Optional[str],
                cover_data: bytes, cover_filename: Optional[str],
                diff_filename: Optional[str]) -> bytes:
        """获取原始图像和隐写图像的差异"""
        try:
            stego_image = ImageUtil.byte_array_to_image(stego_data, stego_filename)
            cover_image = ImageUtil.byte_array_to_image(cover_data, cover_filename)
            
            stego_pixels = ImageUtil.get_image_pixels(stego_image)
            cover_pixels = ImageUtil.get_image_pixels(cover_image)
            
            # 计算差异（放大差异以便可视化）
            diff_pixels = np.abs(stego_pixels.astype(np.int16) - cover_pixels.astype(np.int16)) * 10
            diff_pixels = np.clip(diff_pixels, 0, 255).astype(np.uint8)
            
            diff_image = ImageUtil.set_image_pixels(Image.new('RGB', stego_image.size), diff_pixels)
            return ImageUtil.image_to_byte_array(diff_image, diff_filename)
        except Exception as e:
            raise StegaPyException(str(e), StegaPyErrors.UNHANDLED_EXCEPTION, self.NAMESPACE)
    
    def get_readable_file_extensions(self) -> List[str]:
        """获取支持读取的文件扩展名"""