fixed ShowTransform empty matrix edge case

Josh Klontz
1 parent 2627418b
Showing 2 changed files with 3 additions and 1 deletions
openbr/plugins/gui.cpp
openbr/plugins/slidingwindow.cpp
@@ -447,7 +447,8 @@ public:
             }
             emit this->changeTitle(newTitle);
-            foreach(const cv::Mat & m, t) {
+            foreach (const cv::Mat &m, t) {
+                if (!m.data) continue;
                 qImageBuffer = toQImage(m);
                 displayBuffer->convertFromImage(qImageBuffer);
@@ -270,6 +270,7 @@ class Detector : public Transform
     {
         const float aspectRatio = getAspectRatio(data);
         TemplateList cropped = cropTrainingSamples(data, aspectRatio);
+        qDebug("Detector using: %d training samples.", cropped.size());
         cropped.first().file.set("aspectRatio", aspectRatio);
         transform->train(cropped);
     }