Parcourir la source

Update of SVD metric

jbuisine il y a 5 ans
Parent
commit
e1fff4225d
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  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):