add samples

wangrong

2025-03-19 291afb9c81aae1d15d9e3edcea618d0a2e1fc5a6

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
70
71
72
73
74
75
76
77
78
79
80
81
#include <opencv2/imgproc.hpp> 
 
#include "vp_osd_node_v3.h" 
 
namespace vp_nodes { 
     
    vp_osd_node_v3::vp_osd_node_v3(std::string node_name, std::string font):  
                                vp_node(node_name) { 
        if (!font.empty()) { 
            ft2 = cv::freetype::createFreeType2(); 
            ft2->loadFontData(font, 0);    
        }        
        this->initialized(); 
    } 
     
    vp_osd_node_v3::~vp_osd_node_v3() { 
        deinitialized(); 
    } 
 
    std::shared_ptr<vp_objects::vp_meta> vp_osd_node_v3::handle_frame_meta(std::shared_ptr<vp_objects::vp_frame_meta> meta) { 
       // operations on osd_frame 
        if (meta->osd_frame.empty()) { 
            meta->osd_frame = meta->frame.clone(); 
        } 
 
        auto& canvas = meta->osd_frame; 
        // scan targets 
        for (auto& i : meta->targets) { 
            // track_id 
            auto id = std::to_string(i->track_id); 
            auto labels_to_display = i->primary_label; 
 
            // tracked 
            if (i->track_id != -1) { 
                labels_to_display = "#" + id + " " + labels_to_display; 
            } 
             
            for (auto& label : i->secondary_labels) { 
                labels_to_display += "|" + label; 
            } 
             
            // draw tracks if size>=2 
            if (i->tracks.size() >= 2) { 
                for (int n = 0; n < (i->tracks.size() - 1); n++) { 
                    auto p1 = i->tracks[n].track_point(); 
                    auto p2 = i->tracks[n + 1].track_point(); 
                    cv::line(canvas, cv::Point(p1.x, p1.y), cv::Point(p2.x, p2.y), cv::Scalar(0, 255, 255), 1, cv::LINE_AA); 
                } 
            } 
 
            cv::rectangle(canvas, cv::Rect(i->x, i->y, i->width, i->height), cv::Scalar(255, 255, 0), 2); 
            if (ft2 != nullptr) { 
                ft2->putText(canvas, labels_to_display, cv::Point(i->x, i->y), 20, cv::Scalar(255, 0, 255), cv::FILLED, cv::LINE_AA, true); 
            } 
            else {                
                //cv::putText(canvas, labels_to_display, cv::Point(i->x, i->y), 1, 1, cv::Scalar(255, 0, 255)); 
                int baseline = 0; 
                auto size = cv::getTextSize(labels_to_display, 1, 1, 1, &baseline); 
                vp_utils::put_text_at_center_of_rect(canvas, labels_to_display, cv::Rect(i->x, i->y - size.height, size.width, size.height), true); 
            } 
 
            /* mask area */ 
            if (!i->mask.empty()) { 
                // Resize the mask, threshold, color and apply it on the image 
                cv::resize(i->mask, i->mask, cv::Size(i->width, i->height)); 
                cv::Mat mask = (i->mask > mask_threshold); 
                cv::Mat coloredRoi = (0.3 * cv::Scalar(255, 255, 0) + 0.7 * meta->osd_frame(cv::Rect(i->x, i->y, i->width, i->height))); 
                coloredRoi.convertTo(coloredRoi, CV_8UC3); 
             
                // Draw the contours on the image 
                vector<cv::Mat> contours; 
                cv::Mat hierarchy; 
                mask.convertTo(mask, CV_8U); 
                cv::findContours(mask, contours, hierarchy, cv::RETR_CCOMP, cv::CHAIN_APPROX_SIMPLE); 
                cv::drawContours(coloredRoi, contours, -1, cv::Scalar(255, 255, 0), 5, cv::LINE_8, hierarchy, 100); 
                coloredRoi.copyTo(meta->osd_frame(cv::Rect(i->x, i->y, i->width, i->height)), mask);   
            }                     
        } 
        return meta; 
    } 
}