basic版本的yolo,在yolov3版本上增加人体跟踪
xuepengqiang
2020-05-26 5966f2b095841627d62daac0159e81f83544b85c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
//
// 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;
}