File size: 2,718 Bytes
96d549d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
import cv2
import numpy as np
from abc import ABCMeta, abstractmethod
import operator
import scipy.spatial.distance as dist
class ImageComparator(object):
__metaclass__ = ABCMeta
@abstractmethod
def are_similar(self, first, second):
pass
def __init__(self, threshold):
self.threshold = threshold
def are_same(self, first, second, op=operator.ge):
return op(self.are_similar(first, second), self.threshold)
class AbsDiffHistComparator(ImageComparator):
def __init__(self, threshold):
super(AbsDiffHistComparator, self).__init__(threshold)
def are_similar(self, first, second):
res = cv2.absdiff(first, second)
hist = cv2.calcHist([res], [0], None, [256], [0, 256])
return 1 - np.sum(hist[15::]) / np.sum(hist)
class EuclideanComparator(ImageComparator):
def __init__(self, threshold):
super(EuclideanComparator, self).__init__(threshold)
def are_similar(self, first, second):
return dist.euclidean(first, second)
class ChebysevComparator(ImageComparator):
def __init__(self, threshold):
super(ChebysevComparator, self).__init__(threshold)
def are_similar(self, first, second):
return dist.chebyshev(first, second)
class OpenCVComparator(ImageComparator):
__metaclass__ = ABCMeta
def __init__(self, threshold):
super(OpenCVComparator, self).__init__(threshold)
@abstractmethod
def get_technique(self):
pass
def are_similar(self, first, second):
result = 0
for i in xrange(3):
hist1 = cv2.calcHist([first], [i], None, [256], [0,256])
hist2 = cv2.calcHist([second], [i], None, [256], [0,256])
result += cv2.compareHist(hist1, hist2, self.get_technique())
return result / 3
class CorrelationOpenCVComparator(OpenCVComparator):
def __init__(self, threshold):
super(CorrelationOpenCVComparator, self).__init__(threshold)
def get_technique(self):
return cv2.HISTCMP_CORREL
class ChiOpenCVComparator(OpenCVComparator):
def __init__(self, threshold):
super(ChiOpenCVComparator, self).__init__(threshold)
def get_technique(self):
return cv2.HISTCMP_CHISQR
class IntersectionOpenCVComparator(OpenCVComparator):
def __init__(self, threshold):
super(IntersectionOpenCVComparator, self).__init__(threshold)
def get_technique(self):
return cv2.HISTCMP_INTERSECT
class BhattacharyyaOpenCVComparator(OpenCVComparator):
def __init__(self, threshold):
super(BhattacharyyaOpenCVComparator, self).__init__(threshold)
def get_technique(self):
return cv2.HISTCMP_BHATTACHARYYA
|