Changeset 514

Timestamp:

06/03/09 20:52:39 (3 years ago)

Author:

spenceal

Message:

 

Location:

trunk/software/rb/vision

Files:

• _vision.c (modified) (1 diff)
• classify.c (modified) (1 diff)

trunk/software/rb/vision/_vision.c

@@ -68 +68 @@
 _init(PyObject *self, PyObject *args)
 {
-        int fake_ccam, classifier_num;
+        char fake_ccam, classifier_num;
         if (!PyArg_ParseTuple(args, "bb", &fake_ccam, &classifier_num))
                 return NULL;

trunk/software/rb/vision/classify.c

@@ -84 +84 @@
         classifier->train_desired = cvCreateMat(1, 1, CV_32F);
         switch (type) {
-        case RB_DTREE: classifier->classifier = dtree_create(); break;
-        case RB_HARD:  break;
-        default: fprintf(stderr, "Warning: unknown classifier id %d\n", type);
+        case RB_DTREE:
+                printf("Using decision tree classifier\n");
+                classifier->classifier = dtree_create();
+                break;
+        case RB_HARD:
+                printf("Using hard coded classifier\n");
+                break;
+        default:
+                fprintf(stderr, "Warning: unknown classifier id %d\n", type);
+                break;
         }
         return classifier;