VSAG/amadeus_26_fb
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-09 18:45:00 +08:00
parent cdef78ddde
commit cd5528cfb7
3 changed files with 3 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
armor/src/armor_finder/armor_finder.cpp
View File

@@ -28,9 +28,7 @@ ArmorFinder::ArmorFinder(uint8_t &color, Serial &u, const string &paras_folder,
contour_area(0),
use_classifier(use),
boxid(-1),
tracking_cnt(0),
miss_cnt(0),
weak_cnt(0)
tracking_cnt(0)
{
}
@@ -44,16 +42,8 @@ void ArmorFinder::run(cv::Mat &src) {
// stateSearchingTarget(src_use); // for debug
// return;
switch (state){
case WEAKSEARCHING_STATE:
if(stateSearchingTarget(src_use) && ++weak_cnt>5){
miss_cnt = 0;
state = SEARCHING_STATE;
}else{
weak_cnt = 0;
}
case SEARCHING_STATE:
if(stateSearchingTarget(src_use)){
miss_cnt = 0;
if((armor_box & cv::Rect2d(0, 0, 640, 480)) == armor_box) { // 判断装甲板区域是否脱离图像区域
if(!classifier || !use_classifier){ /* 如果分类器不可用或者不使用分类器 */
cv::Mat roi = src_use.clone()(armor_box), roi_gray; /* 就使用装甲区域亮点数判断是否跟丢 */
@@ -67,14 +57,10 @@ void ArmorFinder::run(cv::Mat &src) {
tracking_cnt = 0;
LOGM(STR_CTR(WORD_LIGHT_CYAN, "into track"));
}
}else if(++miss_cnt>100){
weak_cnt = 0;
state = WEAKSEARCHING_STATE;
}
break;
case TRACKING_STATE:
if(!stateTrackingTarget(src_use) || ++tracking_cnt>100){ // 最多追踪100帧图像
miss_cnt = 0;
state = SEARCHING_STATE;
LOGM(STR_CTR(WORD_LIGHT_YELLOW ,"into search!"));
}

2
armor/src/armor_finder/searching_state/searching_state.cpp
View File

@@ -291,7 +291,7 @@ bool ArmorFinder::stateSearchingTarget(cv::Mat &src) {
showArmorBoxVector("boxes", src, armor_boxes);
cv::waitKey(1);
}
if (classifier && use_classifier && state!=WEAKSEARCHING_STATE) {
if (classifier && use_classifier) {
for (auto box : armor_boxes) {
cv::Mat roi = src(box).clone();
cv::resize(roi, roi, cv::Size(48, 36));