Python OpenCV实现BRISQUE图像质量评估时cv.quality模块无法识别

常见故障点与对应修复方案 #

• 版本不匹配问题
  quality模块从OpenCV 4.1.0版本开始才正式纳入opencv-contrib扩展包，低于该版本无法调用。同时必须保证opencv-python和opencv-contrib-python两个包的版本号完全一致，否则会出现基础功能正常、扩展模块加载失败的情况。
  先执行以下代码确认当前环境的OpenCV版本：

  import cv2
  print(cv2.__version__)
  

  如果版本低于4.1.0，或者两个包版本不一致，先激活你的Anaconda对应虚拟环境，执行以下命令彻底清理冲突包后重装：

  pip uninstall opencv-python opencv-contrib-python opencv-python-headless -y
  conda remove opencv libopencv -y
  pip install opencv-python==4.8.0.74 opencv-contrib-python==4.8.0.74
  

  注意：尽量用pip安装官方预编译的OpenCV包，conda源的opencv-contrib普遍存在模块裁剪不全的问题，容易出现模块缺失。

• 调用逻辑错误
  官方文档标注的cv.quality.QualityBRISQUE.compute()不是可以直接传参调用的静态方法，必须先初始化评估器实例，同时加载BRISQUE算法必需的模型和范围参数文件，否则就算模块存在也会运行报错。
  正确调用步骤：

  1. 提前准备好两个必需的配置文件：brisque_model_live.yml和brisque_range_live.yml，两个文件可以从opencv_contrib源码的testdata/quality目录下获取
  2. 参考以下可运行示例编写代码：

  import cv2 as cv
  
  # 替换为本地配置文件的实际存储路径
  MODEL_PATH = "brisque_model_live.yml"
  RANGE_PATH = "brisque_range_live.yml"
  # 替换为待评估图像路径
  IMG_PATH = "test.jpg"
  
  # 初始化BRISQUE质量评估器
  brisque_evaluator = cv.quality.QualityBRISQUE.create(MODEL_PATH, RANGE_PATH)
  img = cv.imread(IMG_PATH)
  # 计算评分：返回值为元组，第一个值为质量评分，分数越低代表图像质量越好
  quality_score = brisque_evaluator.compute(img)[0]
  print(f"图像BRISQUE质量评分：{quality_score}")
  

• 环境路径冲突
  如果重装包之后依然提示找不到quality模块，检查当前Python环境的包加载路径，确认导入的cv2是你刚安装的pip版本，而不是Anaconda默认路径下自带的旧版本cv2。可以执行以下代码查看cv2的实际加载路径：

import cv2
print(cv2.__file__)

如果路径指向Anaconda根目录的site-packages而不是你当前虚拟环境的site-packages，说明环境激活配置有问题，退出当前环境重新激活，或者在对应虚拟环境里重新执行安装命令即可。

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