From ece612c9683e271ab9677728380c702d2844ebb2 Mon Sep 17 00:00:00 2001
From: xinyang <txy15436@gmail.com>
Date: Tue, 23 Jul 2019 15:29:23 +0800
Subject: [PATCH] fix bug

---
 armor/src/armor_finder/classifier/classifier.cpp | 6 +++++-
 armor/src/armor_finder/find/find_armor_box.cpp   | 2 +-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/armor/src/armor_finder/classifier/classifier.cpp b/armor/src/armor_finder/classifier/classifier.cpp
index 404237c..b8c92b5 100644
--- a/armor/src/armor_finder/classifier/classifier.cpp
+++ b/armor/src/armor_finder/classifier/classifier.cpp
@@ -303,5 +303,9 @@ int Classifier::operator()(const cv::Mat &image) {
 //    cout << result << "==============" <<endl;
     MatrixXd::Index minRow, minCol;
     result.maxCoeff(&minRow, &minCol);
-    return minRow;
+    if(result(minRow, minCol) > 0.9){
+        return minRow;
+    }else{
+        return 0;
+    }
 }
\ No newline at end of file
diff --git a/armor/src/armor_finder/find/find_armor_box.cpp b/armor/src/armor_finder/find/find_armor_box.cpp
index 4a39b96..4fa2589 100644
--- a/armor/src/armor_finder/find/find_armor_box.cpp
+++ b/armor/src/armor_finder/find/find_armor_box.cpp
@@ -150,7 +150,7 @@ bool ArmorFinder::findArmorBox(const cv::Mat &src, ArmorBox &box) {
                 char filename[100];
                 sprintf(filename, PROJECT_DIR"/armor_box_photo/%s_%d.jpg", id2name[one_box.id].data(),
                         time(nullptr) + clock());
-                cv::imwrite(filename, src(armor_box.rect & cv::Rect2d(0, 0, src.rows, src.cols)));
+                cv::imwrite(filename, src(armor_box.rect & cv::Rect2d(0, 0, src.cols, src.rows)));
             }
         }
         if (box.rect == cv::Rect2d(0, 0, 0, 0)) {