如何改进基于OpenCV(Python)的图像双掩码生成代码？

先看看你的效果：
原图像

输出效果

你的代码确实能实现功能，但有几个可以优化的地方，我帮你梳理下：

原代码存在的问题 #

• 直接在原始图像frame上绘制掩码，会破坏原图，后续如果需要复用原图就很麻烦
• 重复计算int(h / 10)，代码冗余且后续修改步长时要改多处
• lower_mask的坐标逻辑有问题：你设置的y起始点比结束点大，OpenCV的rectangle要求第二个点的y坐标必须大于第一个，否则绘制会异常（大概率是笔误）
• 变量命名可以更直观，比如upper_mask改成top_mask更容易理解
• 缺少结果保存或显示的代码，功能流程不完整

优化后的代码 #

import cv2
import numpy as np

# 读取图像，增加读取失败判断
frame = cv2.imread('image.jpg')
if frame is None:
    raise ValueError("无法读取图像，请检查文件路径是否正确")

h, w = frame.shape[:2]
h_step = h // 10  # 提取重复计算的步长，提升可读性和维护性

# 创建与原图同尺寸的独立掩码，初始全黑
mask = np.zeros_like(frame)

# 绘制顶部掩码：从图像顶部到5倍步长位置
top_mask_y_end = h_step * 5
cv2.rectangle(mask, (0, 0), (w, top_mask_y_end), (0, 50, 255), -1)

# 绘制底部掩码：从7倍步长位置到图像底部（修正原代码的坐标错误）
bottom_mask_y_start = h_step * 7
cv2.rectangle(mask, (0, bottom_mask_y_start), (w, h), (0, 50, 255), -1)

# 将掩码与原图叠加（透明度可根据需求调整）
result = cv2.addWeighted(frame, 1, mask, 0.5, 0)

# 显示并保存结果
cv2.imshow('Masked Result', result)
cv2.imwrite('output_masked.png', result)
cv2.waitKey(0)
cv2.destroyAllWindows()

优化说明 #

• 分离掩码与原图：用np.zeros_like创建独立的掩码数组，完全不修改原始图像，保留原图的可用性
• 提取重复计算：把h//10赋值给h_step，后续所有相关计算都复用这个变量，修改步长时只需要改一处
• 修正坐标逻辑：底部掩码的起始y坐标设置为7倍步长（对应原代码的upper_value + h_step*2），结束点设为图像高度h，确保坐标逻辑正确
• 增强鲁棒性：增加了图像读取失败的判断，避免后续代码无意义报错
• 完整功能流程：添加了图像叠加、显示和保存的代码，让整个处理流程闭环
• 可读性提升：变量命名更直观，添加注释说明每个步骤的作用

内容的提问来源于stack exchange，提问作者lucians