Python2-比较当前图片跟图库那个图片相似

Posted dmtz

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Python2-比较当前图片跟图库那个图片相似相关的知识,希望对你有一定的参考价值。

 

# -*- coding: utf-8 -*-
‘‘‘
Created on 2019年7月22日

‘‘‘
from selenium import webdriver
from time import sleep
from PIL import Image
import random
import os
import cv2
import numpy as np



url ="URL"
driver = webdriver.Chrome()
driver.implicitly_wait(10)
driver.maximize_window()
driver.get(url)
sleep(2)


driver.save_screenshot("E:/test/das.png")

p1=rE:/test/das1.png
p2=rE:/test/das2.png
p3=rE:/test/das3.png
p4=rE:/test/das4.png



element = driver.find_element_by_id("imgcode")
left = element.location[x]
top = element.location[y]
right = element.location[x] + element.size[width]
bottom = element.location[y] + element.size[height]
im1 = Image.open(rE:/test/das.png)
im1 = im1.crop((left, top, right, bottom))
im1.save(r"E:/test/dascode.png")


img = Image.open("E:/test/dascode.png")


cropped = img.crop((0, 0, 25, 30))  # (left, upper, right, lower)
cropped.save(p1)
cropped = img.crop((25, 0, 50, 30))  # (left, upper, right, lower)
cropped.save(p2)
cropped = img.crop((50, 0, 75, 30))  # (left, upper, right, lower)
cropped.save(p3)
cropped = img.crop((75, 0, 96, 30))  # (left, upper, right, lower)
cropped.save(p4)


def getGray(image_file):
    tmpls=[]
    for h in range(0, image_file.size[1]):#h
        for w in range(0, image_file.size[0]):#w
            tmpls.append( image_file.getpixel((w,h)) )

    return tmpls

def getAvg(ls):#获取平均灰度值
    return sum(ls)/len(ls)



def aHash(fne):
    image_file = Image.open(fne) # 打开
    image_file=image_file.resize((35,35))#重置图片大小我12px X 12px
    image_file=image_file.convert("L")#转256灰度图
    Grayls=getGray(image_file)#灰度集合
    avg=getAvg(Grayls)#灰度平均值
    bitls=‘‘#接收获取0或1
    for h in range(1, image_file.size[1]-1):#h
        for w in range(1, image_file.size[0]-1):#w
            if image_file.getpixel((w,h))>=avg:#像素的值比较平均值 大于记为1 小于记为0
                bitls=bitls+1
            else:
                bitls=bitls+0
    return bitls

def getMH(i1,i2):
    a=aHash(i1)
    b=aHash(i2)
    dist = 0;
    for i in range(0,len(a)):
        if a[i]==b[i]:
            dist=dist+1
    return dist



def match(a,rootdir):
    list = os.listdir(rootdir) 
    li=[]
    for i in list:
        re=getMH(a,rootdir+"/"+i)
        li.append(re)
    b=str(li.index(max(li))+1)   
    a=li.index(max(li))
    return b,list[a].split(".")[0]


a=match(E:/test/das4.png,"E:/test/pic4")
print a

 

以上是关于Python2-比较当前图片跟图库那个图片相似的主要内容,如果未能解决你的问题,请参考以下文章

某图片网站整合yolo图片特征相似图片获取

比较两张相似的图片得到相似度值

图片相似度——hash算法简介

iOS图片相似度比较

如何通过 C# 比较两幅图片的相似度?

JAVA 比较两张图片的相似度