fix(classifier): rename 'none' to 'raw'

williamfzc · Jul 18, 2019 · 4586826 · 4586826
1 parent 595d051
commit 4586826
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/example/classify_with_svm.py b/example/classify_with_svm.py
@@ -3,7 +3,7 @@
 
 
 # 默认情况下使用 HoG 进行特征提取
-# 你可以将其关闭从而直接对原始图片进行训练与测试：feature_type='none'
+# 你可以将其关闭从而直接对原始图片进行训练与测试：feature_type='raw'
 cl = SVMClassifier(feature_type='hog')
 
 # 基本与SSIM分类器的流程一致

diff --git a/stagesepx/classifier/svm.py b/stagesepx/classifier/svm.py
@@ -17,7 +17,7 @@ class SVMClassifier(BaseClassifier):
         # 'surf': toolbox.turn_surf_desc,
 
         # do not use feature transform
-        'none': lambda x: x,
+        'raw': lambda x: x,
     }
 
     def __init__(self, feature_type: str = None):