直方图比较图像，相似度小于5时重名名

#-*- coding:utf-8 -*-
import cv2
import os
from nt import chdir

class CompareImage(object):

    def __init__(self, image_1_path, image_2_path):
        self.minimum_commutative_image_diff = 0.25
        self.image_1_path = image_1_path
        self.image_2_path = image_2_path
    #返回值为10000图像数据相同
    def compare_image(self):
        image_1 = cv2.imread(self.image_1_path, 0)
        image_2 = cv2.imread(self.image_2_path, 0)
        img_hist_diff, img_template_diff, commutative_image_diff = self.get_image_difference(image_1, image_2)

        if img_hist_diff<0.1 and img_template_diff<0.1:
            if commutative_image_diff < self.minimum_commutative_image_diff:
                return commutative_image_diff
        return 10000 # random failure value

    @staticmethod
    def get_image_difference(image_1, image_2):
        first_image_hist = cv2.calcHist([image_1], [0], None, [256], [0, 256])
        second_image_hist = cv2.calcHist([image_2], [0], None, [256], [0, 256])

        img_hist_diff = 1-cv2.compareHist(first_image_hist, second_image_hist,0)
        img_template_probability_match = cv2.matchTemplate(first_image_hist, second_image_hist, cv2.TM_CCOEFF_NORMED)[0][0]
        img_template_diff = 1 - img_template_probability_match

        # taking only 10% of histogram diff, since it's less accurate than template method
        commutative_image_diff = (img_hist_diff / 5) + img_template_diff
        return [img_hist_diff,img_template_diff,commutative_image_diff]

#打开文件夹
def open_file(file_name):
    for root, dirs, files in os.walk(file_name):
        temp = 0
        for i in range(len(files)):
            #files[i].index('chong')！=-1找到字符串，图像已经比较过
            if files[i].find('f') != -1:
                continue
            # #初始值
            my_file = os.path.join(root, files[i])
            for j in range(len(files)-1):
                # print my_file, '==='
                # 比较两张图相似点，相同则重命名第二张图
                if files[j].find('f')!=-1:
                    continue
                compare_image = CompareImage(os.path.join(root, files[i]), os.path.join(root, files[j+1]))
                image_difference = compare_image.compare_image()
                print image_difference
                if image_difference != 10000:
                    print os.path.join(root, files[j])
                    # 图像相同重命名 148.8_543.6_18020180320233355_180
                    # print os.path.join(root,files[j][0:files[j].find('.jpg')]+"_chong.jpg")
                    pre = os.path.join(root, files[j])
                    # chdir(os.path.dirname(pre))
                    try:
                        os.rename(pre, os.path.join(root, str(temp) + "_chong_") + str(j) + ('.jpg'))
                    except WindowsError:
                        pass
                    # try:
                    #     print os.path.join(root,str(temp)+"_f_")+str(j)+('.jpg')
                    #     os.rename(pre,os.path.join(root,str(temp)+"_f_")+str(j)+('.jpg'))
                    # except WindowsError:
                    #     pass
            temp+=1
if __name__ == '__main__':
    file_name = 'F:\\test\\test'
    open_file(file_name)
    # chdir(file_name+'/1.jpg')