Browse Source

Update of SVD metric

jbuisine 3 years ago
parent
commit
e1fff4225d
1 changed files with 1 additions and 1 deletions
  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):