解锁豆瓣高清海报(二) 使用 OpenCV 拼接和压缩

解锁豆瓣高清海报(二): 使用 OpenCV 拼接和压缩

脚本地址:

项目地址: Gazer

PixelWeaver.py

pixel_squeezer_cv2.py

前瞻

继上一篇“解锁豆瓣高清海报(一) 深度爬虫与requests进阶之路”成功爬取豆瓣电影海报之后，本文将介绍如何使用 OpenCV 对这些海报进行智能拼接和压缩。我们将开发两个实用的 Python 脚本：PixelWeaver.py 和 pixel_squeezer_cv2.py，前者可以根据指定的行列数和目标尺寸，智能地裁剪、调整并拼接多张图片，实现近似无损的拼接效果；后者则可以将大型图片压缩到指定大小，同时尽可能地保持图片质量。这两个脚本也可以作为通用的图片拼接/压缩工具使用。

使用方法

1. 克隆或下载项目代码。
2. 安装依赖: pip install opencv-python numpy ，或者克隆项目代码后 pip install -r requirements.txt

PixelWeaver.py

pixel_squeezer_cv2.py

PixelWeaver.py

代码结构

1. process_image 处理单张图片，进行智能裁剪或调整 (OpenCV)
2. extract_date 从文件名中提取日期，用于排序 (regex)
3. process_all_images 处理目录下的所有图片，返回处理后的图片列表
4. create_collage 拼接海报并输出

说明和测试结果

上一篇中提到了仍有可能碰到 418 错误导致不能成功保存图片, 故建议按照日志中的链接保存图片并规范命名(只要以日期形式如 2024_12_31 开头即可), 以保证这个脚本中的正则表达式能愉快工作.

关于设置目标尺寸 target_size, 可以观察爬取的图片的分辨率, 我测试之后使用的数据是 1080 × 1600. 过大的图片会根据中心位置裁剪, 过小的图片使用插值缩放.

输入图片数量不足以填满拼接网格的处理

当输入的图片数量无法填满 rows * cols 的拼接网格时，create_collage 函数会在空白位置填充黑色。为了避免程序出错, 修改后的代码会在计算 row 或 col 超出预设范围时，自动停止放置图片, 不会尝试访问不存在的图片元素。

原始的 create_collage 函数的核心循环部分：

  for i in range(num_images):
      row = i // cols
      col = i % cols
      x = col * target_w
      y = row * target_h
      collage[y:y+target_h, x:x+target_w] = processed_images[i]

这段代码的逻辑是：依次取出 processed_images 列表中的每一张图片，计算它应该放置的行列位置 (row, col) 和坐标 (x, y)，然后将图片复制到 collage 画布的相应位置。

问题出在哪里呢？

在拼接图片的过程中，如果遇到图片数量 num_images 少于预设的行列数 rows * cols 的情况，原始的代码逻辑可能会导致程序尝试访问 processed_images 列表中不存在的元素，从而引发错误。

为了解决这个问题，修改代码逻辑，加入安全检查机制。

这是修改后的代码：

for i in range(num_images):
    row = i // cols
    col = i % cols
    if row >= rows or col >= cols:
        break  # 超出预设的行列范围，停止放置图片
    x = col * target_w
    y = row * target_h
    collage[y:y+target_h, x:x+target_w] = processed_images[i]

关键在于这行代码：

    if row >= rows or col >= cols:
        break  # 超出预设的行列范围，停止放置图片

它的作用是：在每次循环迭代中，除了根据 i 计算当前图片的 row 和 col 之外，还增加了一个额外的判断：如果计算出来的 row 大于等于预设的 rows，或者 col 大于等于预设的 cols，就立即跳出循环，不再继续放置图片，并结束循环。这确保了循环不会尝试访问 processed_images 列表中不存在的元素，也不会在 collage 画布的错误位置放置图片。

为什么这样可以解决问题？

因为 for i in range(num_images) 循环只会执行 num_images 次，并且 row 和 col 是基于 i 计算得出的，而 if row >= rows or col >= cols: break 这行代码可以确保，即使 num_images 小于 rows * cols，循环也不会尝试访问不存在的图片元素或在超出预设行列范围的位置放置图片。即使 num_images 小于 rows * cols，循环也会在 num_images 次之内结束, 提前结束循环, 避免访问不存在的元素或者在错误位置放置图片。

让我们通过一个例子来说明：

• rows = 3 (预设 3 行)
• cols = 3 (预设 3 列)
• rows * cols = 9 (总共 9 个格子)
• num_images = 7 (实际只有 7 张图片)

在这种情况下，for 循环会执行 7 次 (因为 num_images 是 7)，i 的值依次为 0, 1, 2, 3, 4, 5, 6。

当 i 循环到 6 的时候 (放置第 7 张图片)：

• row = 6 // 3 = 2
• col = 6 % 3 = 0
• row (2) 小于 rows (3), col (0) 小于 cols (3)，所以会放置第 7 张图片 (processed_images[6])

当循环结束时, 不会计算 i=7 时的 row 和 col 了. 因为 num_images 是 7, 所以 for i in range(num_images) 最后一次循环 i 的值是 6.

由于循环在 i=6 时结束, 因此循环不会尝试访问 processed_images[7], 也不会计算 i=7 时的 row 和 col。

这样，即使只有 7 张图片，程序也不会出错，并且会正确地将 7 张图片放置到 collage 画布的前 7 个格子上，剩下的格子将保持空白（在代码中看起来是黑色，因为 collage 初始化为黑色背景）。

总结一下，if row >= rows or col >= cols: break 这行代码在这里起到了一个额外的保护作用。它确保了在 num_images 小于 rows * cols 的情况下，循环不会尝试访问不存在的图片元素或在超出预设行列范围的位置放置图片，从而避免了程序出错。即使我们确定 num_images 小于 rows * cols，这行代码也能增强程序的健壮性，使其能够处理更多不同的情况，并使代码的逻辑更加清晰。

近似无损拼接

PixelWeaver.py 脚本在拼接图片时，通过精心的裁剪和缩放策略，以及高质量的插值算法 (cv2.INTER_AREA 用于缩小, cv2.INTER_CUBIC 用于放大), 尽可能地保留了原始图片的细节, 实现了肉眼难以察觉损失的近似无损拼接。即使是 110 张海报拼接成一张大图，最终的图片质量依然非常出色。

原因：

1. 图片处理过程（process_image 函数）:

   • 在这个脚本中，每张图片都会经过 process_image 函数的处理。这个函数主要做了两件事：

     1. 裁剪: 当图片尺寸大于目标尺寸时，会裁剪图片中心区域。
     2. 缩放: 当图片尺寸不等于目标尺寸时, 会进行缩放, 确保所有图片尺寸一致。

   • 裁剪操作本身就是一种有损操作。因为它会直接丢弃图片边缘的部分像素信息。

   • 缩放操作虽然理论上有可能造成图像信息的损失, 但是, 如果使用高质量的插值算法, 可以将损失控制在肉眼几乎无法察觉的范围内.

2. 拼接过程 (create_collage 函数):
   • 拼接过程本身并不会对图片数据进行修改, 可以看作是无损的. 它只是将所有图片数据按顺序拷贝到一张大的画布上。
3. 图片保存 (cv2.imwrite 函数):
   • cv2.imwrite() 函数在保存图片时, 会根据你指定的文件格式 (例如 .jpg, .png) 进行编码压缩.
   • 如果保存为 .jpg 格式，那么这肯定是一个有损压缩的过程。JPEG 格式是一种有损压缩的图片格式，它通过牺牲一定的图片质量来换取较小的文件体积。
   • 如果保存为 .png 格式，那么可以认为是无损压缩, 或者损失极小。 PNG 格式是一种无损压缩的图片格式，它可以保留图片的原始信息，但文件体积通常比 JPEG 格式大.

所以, 110张海报拼接后得到59.2MB, 并且感觉质量特别棒, 这也印证了我们上面的分析. 因为之前使用的是 .jpg 格式来保存的, 所以图片有压缩, 但是这个压缩带来的损失, 从观感上来说, 是很小的, 几乎无法察觉.

插值算法的功劳:

如果要说哪个算法起到了最大的作用，那应该是在缩放操作中使用的 插值算法。

在代码中，有两种情况会进行缩放：

1. 图片尺寸大于目标尺寸，裁剪后如果尺寸仍然和目标尺寸不同, 则进行缩放：

   if (cropped_img.shape[1] != target_w) or (cropped_img.shape[0] != target_h) :
             cropped_img = cv2.resize(cropped_img, target_size, interpolation = cv2.INTER_AREA)
   

   这里使用了 cv2.INTER_AREA 插值算法。

2. 图片尺寸小于目标尺寸：

       resized_img = cv2.resize(img, target_size, interpolation = cv2.INTER_CUBIC) # 使用 cv2.INTER_CUBIC
   

   这里使用了 cv2.INTER_CUBIC 插值算法。

• cv2.INTER_AREA： 基于局部像素的区域重采样。它适用于缩小图像。这个算法在进行图片缩小的时候, 能较好地保留图片的质量和细节.
• cv2.INTER_CUBIC： 基于 4x4 像素邻域的立方插值。它适用于放大图像。这个算法在图片放大的时候, 能获得一个相对平滑和清晰的结果.

这两个插值算法在 OpenCV 中都属于高质量的插值算法，能够在缩放过程中最大程度地保留图像的细节，减少失真。 因此，即使图片经过了缩放，最终的拼接结果仍然能够保持很好的质量。

process_image 函数详解

process_image 函数负责处理每一张输入图片，它的主要任务是根据目标尺寸 (target_size) 对图片进行智能裁剪或调整。

裁剪策略:

当图片尺寸大于目标尺寸时，为了尽可能保留图片的关键信息，process_image 会裁剪图片的中心区域。这是因为大多数海报的主体内容通常位于图片的中心位置。

缩放策略:

为了保证所有图片都能无缝拼接，process_image 会将所有图片都调整到相同的尺寸 (target_size)。在缩放过程中，根据不同的情况选择不同的插值算法：

# 代码摘自 PixelWeaver.py 的 process_image 函数
if (img.shape[1]  target_w) or (img.shape[0]  target_h):
    # 裁剪图片中心区域
    # ...
    if (cropped_img.shape[1] != target_w) or (cropped_img.shape[0] != target_h) :
        cropped_img = cv2.resize(cropped_img, target_size, interpolation = cv2.INTER_AREA) # 使用 cv2.INTER_AREA 进行缩小
elif (img.shape[1] != target_w) or (img.shape[0] != target_h):
    # 缩放图片以匹配目标尺寸
    resized_img = cv2.resize(img, target_size, interpolation = cv2.INTER_CUBIC) # 使用 cv2.INTER_CUBIC 进行放大

问题来了, 115MB 的图无法分享和上传到网上了, 所以还是使用pixel_squeezer_cv2.py压缩一下吧.

pixel_squeezer_cv2.py

说明和测试结果

使用 OpenCV 压缩 JPEG 图片的参数主要是 cv2.IMWRITE_JPEG_QUALITY，它的范围是 0-100，数值越小，压缩率越高，图片质量越差。

因为不同的图片内容（例如纹理复杂程度、颜色丰富程度）对压缩的敏感度不同。 有些图片可能压缩到 quality=40 还能接受，有些可能到 quality=50 就已经出现明显的失真了。

压缩效果示例 (压缩后为 5.28MB ):

海报压缩效果.jpg

compress_image_cv2 函数详解

compress_image_cv2 函数负责将图片压缩到指定大小 (MB) 以下，同时尽可能地保持图片质量。它接受以下几个关键参数：

• target_size_mb: 目标文件大小，单位为 MB。
• max_width: 图片的最大宽度。如果图片的宽度超过这个值，将会被等比例缩小。
• max_height: 图片的最大高度。如果图片的高度超过这个值，将会被等比例缩小。

工作原理:

compress_image_cv2 函数使用一个 while 循环来不断尝试不同的 JPEG 压缩质量 (quality)，直到压缩后的文件大小小于等于 target_size_mb。

quality = 75  # 初始质量值
while True:
    # ...
    cv2.imwrite(temp_output_path, image, [cv2.IMWRITE_JPEG_QUALITY, quality]) # 使用当前 quality 值保存图片
    file_size_mb = os.path.getsize(temp_output_path) / (1024 * 1024) # 获取文件大小
    print(f"Quality: {quality}, Size: {file_size_mb:.2f} MB")

    if file_size_mb <= target_size_mb:
        # 文件大小已满足要求，跳出循环
        break
    elif quality <= 5:
        # 质量已降至最低，仍然无法满足要求，发出警告并跳出循环
        print("警告：质量已降至极低，可能无法满足目标大小。")
        break
    else:
        quality -= 5  # 降低 quality 值，继续尝试

代码解释:

1. 函数首先根据 max_width 和 max_height 参数调整图片的尺寸。
2. 然后，它从初始的 quality 值 (默认为 75) 开始，在一个 while 循环中不断尝试压缩图片。
3. 每次循环中，它使用当前的 quality 值将图片保存为一个临时文件，并获取临时文件的大小。
4. 如果文件大小小于等于 target_size_mb，则说明压缩成功，函数将临时文件重命名为最终的输出文件，并跳出循环。
5. 如果文件大小仍然大于 target_size_mb，并且 quality 值已经降至 5 或以下，则说明无法在保持可接受质量的前提下将图片压缩到目标大小，函数会发出警告，并将当前质量下的临时文件作为最终输出。
6. 否则，函数会将 quality 值降低 5，并继续循环尝试。
   环中，它使用当前的 quality 值将图片保存为一个临时文件，并获取临时文件的大小。
7. 如果文件大小小于等于 target_size_mb，则说明压缩成功，函数将临时文件重命名为最终的输出文件，并跳出循环。
8. 如果文件大小仍然大于 target_size_mb，并且 quality 值已经降至 5 或以下，则说明无法在保持可接受质量的前提下将图片压缩到目标大小，函数会发出警告，并将当前质量下的临时文件作为最终输出。
9. 否则，函数会将 quality 值降低 5，并继续循环尝试。