init commit

hc

2023-02-13 e440ec23c5a540cdd3f7464e8779219be6fd3d95

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
#include "stdio.h"
#include "objects_update.h"
#include "track_link.h"
 
OdtDetector::OdtDetector()
{  
    printf("success build\n");
}
 
OdtDetector::~OdtDetector()
{
}
 
int OdtDetector::update(int maxTrackLifetime, int track_num_input, object_T* object_input,
                    int* track_num_output, object_T* object_output, int width, int height)
{
 
      m_width = width;
    m_height = height;
 
    std::vector<Rect_T> objects_Rect_T;
    std::vector<int> objects_class;
    std::vector<float> object_score;
    for (int i = 0; i < track_num_input; i++) {
        objects_Rect_T.push_back(object_input[i].r);
        objects_class.push_back(object_input[i].obj_class);
        object_score.push_back(object_input[i].score);
    }
 
    m_objects_tracker.updateTrackedObjects(objects_Rect_T, objects_class, object_score, maxTrackLifetime, m_width, m_height);
    objects_Rect_T.clear();
    objects_class.clear();
    object_score.clear();
 
    std::vector<ObjectsTracker::ExtObject> extObjects;
    m_objects_tracker.getObjects(extObjects);
    
    int nobjects = (int)extObjects.size();
    track_num_output[0] = nobjects;
    
    int i = 0;
    for (; i < nobjects && i < 100; i++)
    {
        if(extObjects[i].miss == 0){
            //object_output[i].r = extObjects[i].location;
            object_output[i].r = extObjects[i].smooth_rect;
 
        }
        else{
            object_output[i].r = extObjects[i].predict_loc_when_miss;
            //object_output[i].score = -10000;
        }
        object_output[i].obj_class = extObjects[i].obj_class;
        object_output[i].score = extObjects[i].score;
        object_output[i].id = extObjects[i].id;
 
    }
    return nobjects;
}