浏览代码

Update of SVD metric

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):