计算两张图片的结构相似性指数SSIM

2024-03-17 21:52

本文主要是介绍计算两张图片的结构相似性指数SSIM,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

结构相似性指数(SSIM)

主要考虑了人眼的感知特性,比较两幅图像的亮度、对比度和结构等特征。SSIM 取值范围在 [-1, 1] 之间,越接近 1 表示两幅图像越相似。比较一个文件中哪些图像存在相似的代码示例:
安装skimage:

pip install scikit-image -i https://mirrors.aliyun.com/pypi/simple/

完整如下:


import os
import sys
import cv2
import shutil
import numpy as np
from tqdm import tqdm
from skimage.metrics import structural_similarity as compare_ssimcurrent_dir = os.path.dirname(os.path.abspath(__file__))
base_dir = current_dir
sys.path.append(base_dir)# 读取一个文件夹的的图片路径列表
def list_files(src_dir, exts, is_dirpath=True):assert exts is not Noneif not isinstance(exts, (str, list, tuple)):raise TypeError('The type of exts must be str or a tuple/list of str, not' % type(exts))exts = [exts] if isinstance(exts, str) else extsfile_list = []for root, dirs, files in os.walk(src_dir):for file in files:if file.lower().endswith(tuple(exts)):if is_dirpath:file_list.append(os.path.normpath(os.path.join(root, file)))else:file_list.append(os.path.normpath(file))return file_list# 读取图片,转为灰度图,缩放为同一个尺寸
def read_ssim_im(fpath):img = np.fromfile(fpath, dtype=np.uint8)img = cv2.imdecode(img, cv2.IMREAD_GRAYSCALE)img = cv2.resize(img, (320, 240))return imgif __name__ == "__main__":src_dir = "D:\\001-数据集\\images"dst_dir = "D:\\001-数据集\\dst_images"os.makedirs(dst_dir, exist_ok=True)# 读取一个images文件夹的图像路径file_paths = list_files(src_dir=src_dir, exts=[".jpg", ".png"])l = len(file_paths)# 读取图像images_data = []for file_path in tqdm(file_paths):images_data.append(read_ssim_im(file_path))# 两两计算,如果相似移动到一个dst_images文件夹中for i in tqdm(range(l)):for j in range(l-i-1):ssim = compare_ssim(images_data[i], images_data[i+j+1], multichannel=True)# print('SSIM:', ssim)if ssim > 0.8:print('=== SSIM:', ssim)shutil.move(src=file_paths[i], dst=os.path.join(dst_dir, os.path.basename(file_paths[i])))shutil.move(src=file_paths[i+j+1], dst=os.path.join(dst_dir, os.path.basename(file_paths[i+j+1])))

如果有用,请 点赞、关注、收藏、分享
你的鼓励是我最大的动力

这篇关于计算两张图片的结构相似性指数SSIM的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/820284

相关文章

关于MongoDB图片URL存储异常问题以及解决

《关于MongoDB图片URL存储异常问题以及解决》:本文主要介绍关于MongoDB图片URL存储异常问题以及解决方案,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐... 目录MongoDB图片URL存储异常问题项目场景问题描述原因分析解决方案预防措施js总结MongoDB图

python实现svg图片转换为png和gif

《python实现svg图片转换为png和gif》这篇文章主要为大家详细介绍了python如何实现将svg图片格式转换为png和gif,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录python实现svg图片转换为png和gifpython实现图片格式之间的相互转换延展:基于Py

使用Python从PPT文档中提取图片和图片信息(如坐标、宽度和高度等)

《使用Python从PPT文档中提取图片和图片信息(如坐标、宽度和高度等)》PPT是一种高效的信息展示工具,广泛应用于教育、商务和设计等多个领域,PPT文档中常常包含丰富的图片内容,这些图片不仅提升了... 目录一、引言二、环境与工具三、python 提取PPT背景图片3.1 提取幻灯片背景图片3.2 提取

Python实现图片分割的多种方法总结

《Python实现图片分割的多种方法总结》图片分割是图像处理中的一个重要任务,它的目标是将图像划分为多个区域或者对象,本文为大家整理了一些常用的分割方法,大家可以根据需求自行选择... 目录1. 基于传统图像处理的分割方法(1) 使用固定阈值分割图片(2) 自适应阈值分割(3) 使用图像边缘检测分割(4)

C#实现将Excel表格转换为图片(JPG/ PNG)

《C#实现将Excel表格转换为图片(JPG/PNG)》Excel表格可能会因为不同设备或字体缺失等问题,导致格式错乱或数据显示异常,转换为图片后,能确保数据的排版等保持一致,下面我们看看如何使用C... 目录通过C# 转换Excel工作表到图片通过C# 转换指定单元格区域到图片知识扩展C# 将 Excel

JS+HTML实现在线图片水印添加工具

《JS+HTML实现在线图片水印添加工具》在社交媒体和内容创作日益频繁的今天,如何保护原创内容、展示品牌身份成了一个不得不面对的问题,本文将实现一个完全基于HTML+CSS构建的现代化图片水印在线工具... 目录概述功能亮点使用方法技术解析延伸思考运行效果项目源码下载总结概述在社交媒体和内容创作日益频繁的

使用Node.js制作图片上传服务的详细教程

《使用Node.js制作图片上传服务的详细教程》在现代Web应用开发中,图片上传是一项常见且重要的功能,借助Node.js强大的生态系统,我们可以轻松搭建高效的图片上传服务,本文将深入探讨如何使用No... 目录准备工作搭建 Express 服务器配置 multer 进行图片上传处理图片上传请求完整代码示例

基于Python实现高效PPT转图片工具

《基于Python实现高效PPT转图片工具》在日常工作中,PPT是我们常用的演示工具,但有时候我们需要将PPT的内容提取为图片格式以便于展示或保存,所以本文将用Python实现PPT转PNG工具,希望... 目录1. 概述2. 功能使用2.1 安装依赖2.2 使用步骤2.3 代码实现2.4 GUI界面3.效

Python实现AVIF图片与其他图片格式间的批量转换

《Python实现AVIF图片与其他图片格式间的批量转换》这篇文章主要为大家详细介绍了如何使用Pillow库实现AVIF与其他格式的相互转换,即将AVIF转换为常见的格式,比如JPG或PNG,需要的小... 目录环境配置1.将单个 AVIF 图片转换为 JPG 和 PNG2.批量转换目录下所有 AVIF 图

详解如何通过Python批量转换图片为PDF

《详解如何通过Python批量转换图片为PDF》:本文主要介绍如何基于Python+Tkinter开发的图片批量转PDF工具,可以支持批量添加图片,拖拽等操作,感兴趣的小伙伴可以参考一下... 目录1. 概述2. 功能亮点2.1 主要功能2.2 界面设计3. 使用指南3.1 运行环境3.2 使用步骤4. 核