jbuisine 6 лет назад
Родитель
Сommit
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):