test code

Scheaven

2021-07-28 eb98b4028b6d6997a458d2ae7ec1635c4d1e4a6c

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
    //
// Created by Scheaven on 2019/11/18.
//
#include "src/h_interface.h"
#include "src/std_target.h"
#include <opencv2/opencv.hpp>           // C++
#include "src/utils/draw_util.h"
 
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;
 
    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;
    cap = VideoCapture("/data/disk2/01_dataset/03_human/34983128.mp4");
 
    while(1){        /* code */
        cap >> frame;
        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 << "confidence:" << t_result->targets[i].confidence << endl;
                // draw_SDK_result(frame, t_result->targets[i]);
            }
        }else{
            cout << "------------over--" << endl;
            release(handle);
            release_result(sr);
            break;
        }
    }
    return 0;
}