【代码篇】图像预处理阶段：提取图像特征

最新推荐文章于 2024-06-28 15:30:07 发布

王纯月

最新推荐文章于 2024-06-28 15:30:07 发布

阅读量1.1k

点赞数

分类专栏： OpenCV python 文章标签： opencv python 计算机视觉

本文链接：https://blog.csdn.net/weixin_42326479/article/details/128434317

版权

本文介绍了在图像预处理阶段如何使用Python和OpenCV库提取图像特征，包括文件大小、尺寸、图像梯度、Sobel算子应用、色彩丰富度和无参考图像质量评价指标NIQE。这些特征有助于进行图像分析和去重操作。

摘要由CSDN通过智能技术生成

前言

需要提取的图像特征：
1.文件大小；2.宽；3.高；4.图片尺寸；5.图像梯度（表征图像纹理的复杂程度）；6.sobel算子（表征图像复杂度）；7.色彩丰富度；8.无参考图像评价指标NIQE。

最后根据上述图像信息可对相同的图片进行去重操作

代码

# *encoding=utf-8
import re
import os
import cv2
import math
import numpy as np
import pandas as pd
import skvideo.measure


# sobel算子
def SI_IMG(img):
    sobelx = cv2.Sobel(img, cv2.CV_64F, dx=1, dy=0)
    sobelx = cv2.convertScaleAbs(sobelx).astype('float32')

    sobely = cv2.Sobel(img, cv2.CV_64F, dx=0, dy=1)
    sobely = cv2.convertScaleAbs(sobely).astype('float32')

    a = sobelx * sobelx + sobely * sobely
    result = np.sqrt(a).astype('float32')

    stddv = result