Increase RGB vector space

Author: Defcon27

FeatureVectors.py

@@ -1,11 +1,14 @@
 import cv2
 import numpy as np
+from Image_Filters.NoiseReduction import NoiseReduction
+from Image_Filters.ConvolutionalFilters import ConvolutionFilter
 
 
 class FeatureVectors():
 
     def __init__(self, image):
-        self.image = image
+        self.filter = NoiseReduction(image)
+        self.image = self.filter.applyGaussianBlur()
 
     def __getMeanIntensity(self):
         meanIntensity = []
@@ -25,12 +28,19 @@ def __getStdIntensity(self):
 
     def __getRGBHistogramVector(self):
         histogram_3d = cv2.calcHist([self.image], [0, 1, 2], None,
-                                    [8, 8, 8], [0, 256, 0, 256, 0, 256])
+                                    [12, 12, 12], [0, 256, 0, 256, 0, 256])
         histogram_3d = histogram_3d.ravel()
         RGBHistogram = list(histogram_3d)
 
         return RGBHistogram
 
+    def __getHuMoments(self):
+        filter = ConvolutionFilter(self.image)
+        canny_filtered = filter.applyCannyEdge()
+        canny_huMoments = cv2.HuMoments(cv2.moments(canny_filtered)).flatten()
+        huVector = list(canny_huMoments.ravel())
+        return huVector
+
     def getFeatureVector(self):
         featureVectors = []
         meanIntensity = self.__getMeanIntensity()

QuerySearch.py

@@ -41,7 +41,7 @@ def performSearch(self):
         for image in self.features:
             queryVector = self.queryFeature
             vector = self.features[image]
-            cosine_similarity = self.__SAD_distance(queryVector, vector)
+            cosine_similarity = self.__chi2_distance(queryVector, vector)
 
             searchSimilarityScores.append((image, cosine_similarity))