Project Icon

pillow-simd

针对x86架构优化的高性能图像处理库

Pillow-SIMD是针对x86架构优化的Pillow库版本,适用于Intel和AMD CPU。通过SIMD计算技术,它显著提升了图像处理性能,部分操作比原版Pillow快16-40倍。支持SSE4和AVX2指令集,可直接替代同版本Pillow。该项目已在生产环境验证,获得Uploadcare支持。Pillow-SIMD优化了多种图像操作,包括调整大小、模糊和合成等。

Pillow-SIMD

Pillow-SIMD 是 Pillow 库 针对 x86 架构(主要是 Intel 和 AMD CPU)高度优化的版本。

Pillow-SIMD "跟随"Pillow,这意味着它可以直接替代相同版本的 Pillow。例如,Pillow-SIMD 3.2.0.post3 可以直接替代 Pillow 3.2.0,而 Pillow-SIMD 3.3.3.post0 可以替代 Pillow 3.3.3

有关原始 Pillow 的更多信息,请参阅: 阅读文档查看更新日志了解如何贡献

为什么选择 SIMD

有多种方法可以提高图像处理性能。举几个例子,这些方法可以是:使用更好的算法、优化现有实现、使用更多的处理能力和/或资源。使用更高效算法的一个很好的例子是基于卷积的高斯模糊替换为序列盒式模糊。

但这样的例子很少见。众所周知,某些处理可以通过并行处理来优化相应的例程。但更实用的优化关键可能是利用现有资源使事情运行得更快。例如,SIMD 计算就是这种情况。

SIMD 代表"单指令多数据",其本质是通过使用多个处理元素同时对多个数据点执行相同的操作。常见的 CPU SIMD 指令集有 MMX、SSE-SSE4、AVX、AVX2、AVX512、NEON。

目前,Pillow-SIMD 可以编译支持 SSE4(默认)或 AVX2。

状态

Pillow-SIMD 项目已经可以用于生产环境。该项目得到了 Uploadcare 的支持,Uploadcare 是一个基于云的图像存储和处理 SAAS。

Uploadcare

事实上,Uploadcare 自 2015 年以来一直在运行 Pillow-SIMD。

以下图像操作目前已经通过 SIMD 加速:

  • 调整大小(基于卷积的重采样):SSE4、AVX2
  • 高斯和盒式模糊:SSE4
  • Alpha 合成:SSE4、AVX2
  • RGBA → RGBa(alpha 预乘):SSE4、AVX2
  • RGBa → RGBA(alpha 除法):SSE4、AVX2
  • RGB → L(灰度):SSE4
  • 3x3 和 5x5 核心滤波器:SSE4、AVX2
  • 分离和获取通道:SSE4

基准测试

Pillow 性能页面上可以找到大量测试。其中包括针对不同版本的 Pillow 和 Pillow-SIMD 以及 ImageMagick、Skia、OpenCV 和 IPP 的基准测试。

结果显示,对于调整大小,Pillow 总是比 ImageMagick 快,而 Pillow-SIMD 则比原始 Pillow 快 4-6 倍。总的来说,使用 AVX2 的 Pillow-SIMD 始终比 ImageMagick 快 16 到 40 倍,并且性能优于 Chromium 中使用的高速图形库 Skia。

为什么 Pillow 本身如此快速

没有使用任何作弊手段。我们在基准测试中使用了相同的高质量调整大小和模糊方法。不同库产生的结果几乎完全一致。测量速率的差异仅由每个算法的性能决定。

为什么 Pillow-SIMD 更快

当然是因为 SIMD 计算。但还有更多原因:大量循环展开,以及标量数据类型无法使用的特定指令。

为什么不将 SIMD 贡献给原始 Pillow

这并不那么简单。首先,原始 Pillow 支持大量架构,而不仅仅是 x86。即使对于 x86 平台,Pillow 通常也是通过预编译二进制文件分发的。为了将 SIMD 集成到预编译的二进制文件中,我们需要执行运行时 CPU 功能检查。要以这种方式编译代码,我们需要向编译器传递 -mavx2 选项。但是使用这个选项,编译器会将 AVX 指令注入到 SSE 函数中(即交换它们),因为每个 SSE 指令都有对应的 AVX 等效指令。因此,编译这样的库没有简单的方法,特别是使用 setuptools。

安装

如果已安装原始 Pillow 的副本,需要先用 $ pip uninstall -y pillow 将其删除。请为您的平台安装先决条件。安装本身很简单,只需运行 $ pip install pillow-simd,如果您使用的是支持 SSE4 的 CPU,一切应该顺利运行。如果您想安装支持 AVX2 的版本,需要向 C 编译器传递额外的标志。最简单的方法是在编译过程中定义 CC 变量。

$ pip uninstall pillow
$ CC="cc -mavx2" pip install -U --force-reinstall pillow-simd

为 Pillow-SIMD 做贡献

请注意,Pillow-SIMD 和 Pillow 是两个独立的项目。请将与 SIMD 无关的错误和改进提交到原始 Pillow。所有对原始 Pillow 的错误修复随后将自动转移到下一个 Pillow-SIMD 版本。

项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号