initial commit

wangrong

2025-01-03 2c4b8838f89bebca97615c5f0bac2c58e0349771

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
#pragma once 
 
#include "../vp_primary_infer_node.h" 
 
namespace vp_nodes { 
    // general image restoration node using Real-ESRGAN 
    // used to enhance quality of frames in video， see more: https://github.com/xinntao/Real-ESRGAN 
    class vp_restoration_node: public vp_primary_infer_node 
    {    
    private: 
        /* onnx network using opencv::dnn as backend */ 
        cv::dnn::Net restoration_net; 
        bool restoration_to_osd = true; 
    protected: 
        // we need a totally new logic for the whole infer combinations 
        // no separate step pre-defined needed in base class 
        virtual void run_infer_combinations(const std::vector<std::shared_ptr<vp_objects::vp_frame_meta>>& frame_meta_with_batch) override; 
        // override pure virtual method, for compile pass 
        virtual void postprocess(const std::vector<cv::Mat>& raw_outputs, const std::vector<std::shared_ptr<vp_objects::vp_frame_meta>>& frame_meta_with_batch) override; 
    public: 
        vp_restoration_node(std::string node_name,  
                            std::string realesrgan_bg_restoration_model, 
                            std::string face_restoration_model = "", 
                            bool restoration_to_osd = true); 
        ~vp_restoration_node(); 
    }; 
}