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
#pragma once
 
#include <string>
#include "../vp_node.h"
 
namespace vp_nodes {
    class vp_seg_osd_node: public vp_node
    {
    private:
        /* data */
        int gap = 60;
        
        // classs names of semantic segmentation
        std::vector<std::string> classes;
        // colors of semantic segmentation
        std::vector<cv::Vec3b> colors;
        void colorizeSegmentation(const cv::Mat &score, cv::Mat &segm);
        void showLegend(cv::Mat& board);
 
    protected:
        virtual std::shared_ptr<vp_objects::vp_meta> handle_frame_meta(std::shared_ptr<vp_objects::vp_frame_meta> meta) override;
    public:
        vp_seg_osd_node(std::string node_name, std::string classes_file = "", std::string colors_file = "");
        ~vp_seg_osd_node();
    };
}