//
|
// Created by Scheaven on 2019/11/18.
|
//
|
#include "h_interface.h"
|
#include "std_target.h"
|
#include "time.h"
|
|
|
#ifdef OPENCV
|
#include <opencv2/opencv.hpp> // C++
|
#pragma comment(lib, "opencv_core249.lib")
|
#pragma comment(lib, "opencv_imgproc249.lib")
|
#pragma comment(lib, "opencv_highgui249.lib")
|
|
#endif // OPENCV
|
using namespace std;
|
using namespace cv;
|
void* handle;
|
|
int main(int argc, char *argv[])
|
{
|
// //视频流信息
|
VideoCapture cap;
|
VideoCapture cap2;
|
int cam_id = 0;
|
int cam_id2 = 1;
|
std::string cam_name = "18eadadc-f583-4517-8c5d-e2a8cb605f97";
|
std::string cam_name2 = "71ddf5e0-c059-48d7-8a58-dfe824d6e462";
|
|
try {
|
// cap.open("rtsp://admin:a1234567@192.168.5.35:554/h264/ch1/main/av_stream");
|
// cap2.open("rtsp://admin:a1234567@192.168.5.34:554/h264/ch1/main/av_stream");
|
cap.open("/data/disk1/project/data/01_reid/sun_37.avi");
|
cap2.open("/data/disk1/project/data/01_reid/sun_51.avi");
|
// cap.open("/home/disk2/workspace/Scheaven/data/01_reid/cD3.avi");
|
// cap2.open("/home/disk2/workspace/Scheaven/data/01_reid/cT3.avi");
|
}catch(exception){
|
cout<<"输入视频"<<endl;
|
return 0;
|
}
|
|
Mat frame, blob;
|
Mat frame2, blob2;
|
|
const char *conf = "../config.json";
|
int *max_chan;
|
int x = 5;
|
max_chan=&x;
|
handle = create(conf, max_chan);
|
|
void *sr;
|
void *sr2;
|
TImage *img = new TImage();
|
TImage *img2 = new TImage();
|
int i = 0;
|
|
while (true)
|
{
|
std::vector<int> cam_ids;
|
//读取视频帧
|
cap >> frame;
|
//cap2 >> frame2;
|
//if(i++%5!= 0)
|
// sleep(1000);
|
|
clock_t t_strat2 = clock();
|
if(!frame.empty())
|
{
|
img->width = frame.cols;
|
img->height = frame.rows;
|
img->channel = frame.channels();
|
img->data = frame.data;
|
sr = get_result(handle, img, 0);
|
|
TResult * t_result = (TResult*) sr;
|
|
cout << "=======t_result->count==" << t_result->count << endl;
|
for (int i=0; i<t_result->count; i++)
|
cout << "====1111111111===confidence:" << t_result->targets[i].confidence << endl;
|
|
}else{
|
cout << "-----------------------over--" << endl;
|
release(handle);
|
release_result(sr);
|
break;
|
}
|
|
clock_t t_strat3 = clock();
|
cout << "rps---"<< CLOCKS_PER_SEC/(t_strat3 - t_strat2) <<endl;
|
//
|
// if(!frame2.empty())
|
// {
|
// img2->width = frame2.cols;
|
// img2->height = frame2.rows;
|
// img2->channel = frame2.channels();
|
// img2->data = frame2.data;
|
//
|
// sr2 = get_result(handle, img, 1);
|
//
|
// TResult * t_result2 = (TResult*) sr2;
|
//
|
// cout << "=======t_result2->count==" << t_result2->count << endl;
|
// for (int i=0; i<t_result2->count; i++)
|
// cout << "====22222222===confidence:" << t_result2->targets[i].id << endl;
|
//
|
// }else{
|
// cout << "-----------------------over2--" << endl;
|
// release(handle);
|
// release_result(sr2);
|
// break;
|
// }
|
}
|
return 0;
|
}
|
|
