//
|
// Created by Scheaven on 2019/11/18.
|
//
|
#include "h_interface.h"
|
#include "std_target.h"
|
#include "tensorflow/core/public/session.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;
|
|
|
int main(int argc, char *argv[])
|
{
|
// //视频流信息
|
VideoCapture cap;
|
VideoCapture cap2;
|
int cam_id = 1;
|
int cam_id2 = 2;
|
|
try {
|
cap.open("../data/cs01.avi");
|
cap2.open("../data/cs02.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;
|
void* handle = create(conf, max_chan);
|
|
void *sr;
|
TImage *img = new TImage();
|
|
|
while (true) {
|
std::vector<int> cam_ids;
|
//读取视频帧
|
cap >> frame;
|
cap2 >> frame2;
|
if(!frame.empty())
|
{
|
img->width = frame.cols;
|
img->height = frame.rows;
|
img->channel = frame.channels();
|
img->data = frame.data;
|
sr = get_result(handle, img, cam_id);
|
TResult * t_result = (TResult*) sr;
|
|
}else{
|
release(handle);
|
release_result(sr);
|
break;
|
}
|
}
|
return 0;
|
}
|