This commit is contained in:
JiatongSun
2019-04-27 17:14:50 +08:00
parent 27d88c8cee
commit e3fc7cf05c
2 changed files with 8 additions and 8 deletions

View File

@@ -11,7 +11,7 @@ using std::vector;
void Energy::StructingElementClose(cv::Mat &src){ void Energy::StructingElementClose(cv::Mat &src){
if (src.empty())return; if (src.empty())return;
//threshold(src, src, energy_part_param_.CAMERA_GRAY_THRESH, 255, THRESH_BINARY); //threshold(src, src, energy_part_param_.CAMERA_GRAY_THRESH, 255, THRESH_BINARY);
Mat element = getStructuringElement(MORPH_RECT, Size(10, 10)); Mat element = getStructuringElement(MORPH_RECT, Size(6, 6));
morphologyEx(src, src, MORPH_CLOSE, element); morphologyEx(src, src, MORPH_CLOSE, element);
} }

View File

@@ -49,15 +49,15 @@ int main(int argc, char *argv[])
} }
WrapperHead *video_armor; WrapperHead *video_armor;
WrapperHead *video_enegy; WrapperHead *video_energy;
if(from_camera) { if(from_camera) {
video_armor = new CameraWrapper("armor"); video_armor = new CameraWrapper("armor");
video_enegy = new CameraWrapper("energy"); video_energy = new CameraWrapper("energy");
}else { }else {
video_armor = new VideoWrapper("/home/xinyang/Desktop/Video.mp4"); video_armor = new VideoWrapper("r_l_640.avi");
video_enegy = new VideoWrapper("/home/xinyang/Desktop/Video.mp4"); video_energy = new VideoWrapper("r_l_640.avi");
} }
if (video_enegy->init() && video_armor->init()) { if (video_energy->init() && video_armor->init()) {
cout << "Video source initialization successfully." << endl; cout << "Video source initialization successfully." << endl;
} }
@@ -69,7 +69,7 @@ int main(int argc, char *argv[])
energy.setAllyColor(ally_color); energy.setAllyColor(ally_color);
energy.setRotation(energy_part_rotation); energy.setRotation(energy_part_rotation);
while (video_armor->read(energy_src) && video_armor->read(armor_src)) while (video_energy->read(energy_src) && video_armor->read(armor_src))
{ {
if(show_origin) { if(show_origin) {
imshow("enery src", energy_src); imshow("enery src", energy_src);
@@ -90,7 +90,7 @@ int main(int argc, char *argv[])
break; break;
} }
} }
delete video_enegy, video_armor; delete video_energy, video_armor;
cout << "Program fails. Restarting" << endl; cout << "Program fails. Restarting" << endl;
} }