VSAG/vision_sjtu_19
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
xinyang
2019-07-23 15:29:23 +08:00
parent 5b90504988
commit ece612c968
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
armor/src/armor_finder/classifier/classifier.cpp
View File

@@ -303,5 +303,9 @@ int Classifier::operator()(const cv::Mat &image) {
// cout << result << "==============" <<endl; // cout << result << "==============" <<endl;
MatrixXd::Index minRow, minCol; MatrixXd::Index minRow, minCol;
result.maxCoeff(&minRow, &minCol); result.maxCoeff(&minRow, &minCol);
return minRow; if(result(minRow, minCol) > 0.9){
return minRow;
}else{
return 0;
}
} }

2
armor/src/armor_finder/find/find_armor_box.cpp
View File

@@ -150,7 +150,7 @@ bool ArmorFinder::findArmorBox(const cv::Mat &src, ArmorBox &box) {
char filename[100]; char filename[100];
sprintf(filename, PROJECT_DIR"/armor_box_photo/%s_%d.jpg", id2name[one_box.id].data(), sprintf(filename, PROJECT_DIR"/armor_box_photo/%s_%d.jpg", id2name[one_box.id].data(),
time(nullptr) + clock()); 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)) { if (box.rect == cv::Rect2d(0, 0, 0, 0)) {