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

更新CNN参数，使用CNN判断tracking是否跟丢。

Browse Source
This commit is contained in:
xinyang
2019-04-30 20:20:35 +08:00
parent e3098fe3fa
commit ee83a373d3
16 changed files with 12416 additions and 9793 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
main.cpp
View File

@@ -15,7 +15,7 @@
#include <options/options.h>
#include <thread>
#define DO_NOT_CNT_TIME
//#define DO_NOT_CNT_TIME
#include <log.h>
#define PATH PROJECT_DIR
@@ -52,13 +52,13 @@ int main(int argc, char *argv[]){
WrapperHead *video_armor;
WrapperHead *video_energy;
if(from_camera) {
// video_armor = new CameraWrapper();
video_energy = new CameraWrapper();
video_armor = new CameraWrapper(0);
// video_energy = new CameraWrapper(1);
}else {
video_armor = new VideoWrapper("/home/xinyang/Desktop/Video.mp4");
video_energy = new VideoWrapper("/home/xinyang/Desktop/Video.mp4");
}
if (video_energy->init()) {
if (video_armor->init()) {
cout << "Video source initialization successfully." << endl;
}
@@ -70,28 +70,32 @@ int main(int argc, char *argv[]){
energy.setAllyColor(ally_color);
energy.setRotation(energy_part_rotation);
while (video_energy->read(energy_src) && video_energy->read(armor_src))
{
if(show_origin) {
imshow("enery src", energy_src);
imshow("armor src", armor_src);
}
if(state == ENERGY_STATE){
if(from_camera==0){
energy.extract(energy_src);
bool ok=true;
while (ok){
CNT_TIME(WORD_LIGHT_CYAN, "Total", {
ok = video_armor->read(energy_src) && video_armor->read(armor_src);
if (show_origin) {
imshow("enery src", energy_src);
imshow("armor src", armor_src);
}
energy.run(energy_src);
}else{
CNT_TIME(WORD_LIGHT_BLUE, "Armor Time", {
armorFinder.run(armor_src);
});
}
if (waitKey(1) == 'q') {
flag = false;
break;
}
if (state == ENERGY_STATE) {
if (from_camera == 0) {
energy.extract(energy_src);
}
energy.run(energy_src);
} else {
CNT_TIME(WORD_LIGHT_BLUE, "Armor Time", {
armorFinder.run(armor_src);
});
}
if (waitKey(1) == 'q') {
flag = false;
break;
}
});
}
delete video_energy;
delete video_armor;
cout << "Program fails. Restarting" << endl;
}