SCAN THE CODE: refactor classifier using base class main function

perception/navigator_vision/navigator_vision/scan_the_code_classifier.py

@@ -10,24 +10,16 @@ class ScanTheCodeClassifier(GaussianColorClassifier):
     def __init__(self):
         rospack = RosPack()
         path = rospack.get_path('navigator_vision')
-        self.features_file = os.path.join(path, 'config/stc_colors.csv')
-        super(ScanTheCodeClassifier, self).__init__(ScanTheCodeClassifier.CLASSES)
-
-    def train_from_csv(self):
-        return super(ScanTheCodeClassifier, self).train_from_csv(self.features_file)
-
-    def save_csv(self, features, classes):
-        return super(ScanTheCodeClassifier, self).save_csv(features, classes, filename=self.features_file)
+        training_file = os.path.join(path, 'config/stc/training.csv')
+        labelfile = os.path.join(path, 'config/stc/labels.json')
+        super(ScanTheCodeClassifier, self).__init__(ScanTheCodeClassifier.CLASSES,
+                                                    training_file=training_file, labelfile=labelfile)
 
 
 if __name__ == '__main__':
     '''
-    When run as an executable, saves the training features to a csv file
-    2 arguemnts: labelbox.io labelfile, and image directory
+    Can be run as executable to extract features or check accuracy score
     '''
     import sys
-    labelfile = sys.argv[1]
-    image_dir = sys.argv[2]
     s = ScanTheCodeClassifier()
-    features, classes = s.extract_labels(labelfile, image_dir)
-    s.save_csv(features, classes)
+    s.main(sys.argv[1:])