jbuisine преди 6 години
родител
ревизия
e1fff4225d
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      modules/image_metrics/svd_metric.py

+ 1 - 1
modules/image_metrics/svd_metric.py

@@ -11,7 +11,7 @@ Method which extracts SVD features from image and returns 's' vector
 def get_s_model_data(image):
     U, s, V = svd(image, full_matrices=False)
     size = len(s)
-    result = s.reshape([size, 1, 1]) # one shape per canal
+    result = s.reshape([size, 1, 3]) # one shape per canal
     return result
 
 def get(image):