training bug fix

Josh Klontz
1 parent 0ffc0a8f
Showing 2 changed files with 5 additions and 1 deletions
sdk/openbr_plugin.cpp
sdk/openbr_plugin.h
@@ -868,6 +868,7 @@ void MatrixOutput::initialize(const FileList &amp;targetFiles, const FileList &amp;query
     data.create(queryFiles.size(), targetFiles.size(), CV_32FC1);
 }
  
+/* MatrixOutput - protected methods */
 QString MatrixOutput::toString(int row, int column) const
 {
     if (targetFiles[column] == "Label")
@@ -881,6 +882,8 @@ void MatrixOutput::set(float value, int i, int j)
     data.at<float>(i,j) = value;
 }
  
+BR_REGISTER(Output, MatrixOutput)
+
 /* Gallery - public methods */
 TemplateList Gallery::read()
 {
@@ -1164,7 +1167,7 @@ void Distance::train(const TemplateList &amp;templates)
 {
     const TemplateList samples = templates.mid(0, 2000);
     const QList<float> sampleLabels = samples.labels<float>();
-    QSharedPointer<MatrixOutput> memoryOutput((MatrixOutput*)Output::make("Matrix", FileList(samples.size()), FileList(samples.size())));
+    QScopedPointer<MatrixOutput> memoryOutput(dynamic_cast<MatrixOutput*>(Output::make(".Matrix", FileList(samples.size()), FileList(samples.size()))));
     compare(samples, samples, memoryOutput.data());
  
     double genuineAccumulator, impostorAccumulator;
@@ -703,6 +703,7 @@ protected:
  
         const QString abstraction = baseClassName();
         if (name.endsWith(abstraction)) name = name.left(name.size()-abstraction.size());
+        if (name.startsWith("br::")) name = name.right(name.size()-4);
         if (registry->contains(name)) qFatal("%s registry already contains object named: %s", qPrintable(abstraction), qPrintable(name));
         registry->insert(name, this);
     }