123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198 |
- #ifndef MODULES_INFERENCE_INCLUDE_POSTPROC_HPP_
- #define MODULES_INFERENCE_INCLUDE_POSTPROC_HPP_
- #include <memory>
- #include <string>
- #include <unordered_map>
- #include <utility>
- #include <vector>
- #include "easyinfer/model_loader.h"
- #include "reflex_object.h"
- #include "cnstream_frame.hpp"
- #include "cnstream_frame_va.hpp"
- namespace cnstream {
- class Postproc : virtual public ReflexObjectEx<Postproc> {
- public:
-
- virtual ~Postproc() = 0;
-
- static Postproc* Create(const std::string& proc_name);
-
- virtual bool Init(const std::unordered_map<std::string, std::string> ¶ms) { return true; }
-
- void SetThreshold(const float threshold);
-
- virtual int Execute(const std::vector<float*>& net_outputs, const std::shared_ptr<edk::ModelLoader>& model,
- const CNFrameInfoPtr& package) { return 0; }
-
- virtual int Execute(const std::vector<void*>& net_outputs, const std::shared_ptr<edk::ModelLoader>& model,
- const std::vector<CNFrameInfoPtr> &packages) { return 0; }
- protected:
- float threshold_ = 0;
- };
- class ObjPostproc : virtual public ReflexObjectEx<ObjPostproc> {
- public:
-
- virtual ~ObjPostproc() = 0;
-
- static ObjPostproc* Create(const std::string& proc_name);
-
- virtual bool Init(const std::unordered_map<std::string, std::string> ¶ms) { return true; }
-
- void SetThreshold(const float threshold);
-
- virtual int Execute(const std::vector<float*>& net_outputs, const std::shared_ptr<edk::ModelLoader>& model,
- const CNFrameInfoPtr& finfo, const std::shared_ptr<CNInferObject>& pobj) { return 0; }
-
- virtual int Execute(const std::vector<void*>& net_outputs, const std::shared_ptr<edk::ModelLoader>& model,
- const std::vector<std::pair<CNFrameInfoPtr, std::shared_ptr<CNInferObject>>>& obj_infos) {
- return 0;
- }
- protected:
- float threshold_ = 0;
- };
- }
- #endif
|