123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990 |
- #ifndef EDK_SAMPLES_CNPOSTPROC_H_
- #define EDK_SAMPLES_CNPOSTPROC_H_
- #include <string>
- #include <utility>
- #include <vector>
- #include "cxxutil/exception.h"
- #include "easytrack/easy_track.h"
- namespace edk {
- class CnPostproc {
- public:
- virtual ~CnPostproc() {}
- void set_threshold(const float threshold);
-
- std::vector<DetectObject> Execute(const std::vector<std::pair<float*, uint64_t>>& net_outputs);
- protected:
-
- virtual std::vector<DetectObject> Postproc(const std::vector<std::pair<float*, uint64_t>>& net_outputs) = 0;
- float threshold_ = 0;
- };
- class ClassificationPostproc : public CnPostproc {
- protected:
- std::vector<DetectObject> Postproc(const std::vector<std::pair<float*, uint64_t>>& net_outputs) override;
- };
- class SsdPostproc : public CnPostproc {
- protected:
- std::vector<DetectObject> Postproc(const std::vector<std::pair<float*, uint64_t>>& net_outputs) override;
- };
- class Yolov3Postproc : public CnPostproc {
- public:
- inline void set_padl_ratio(float ratio) { padl_ratio_ = ratio; }
- inline void set_padb_ratio(float ratio) { padb_ratio_ = ratio; }
- inline void set_padr_ratio(float ratio) { padr_ratio_ = ratio; }
- inline void set_padt_ratio(float ratio) { padt_ratio_ = ratio; }
- inline float padl_ratio() const { return padl_ratio_; }
- inline float padb_ratio() const { return padb_ratio_; }
- inline float padr_ratio() const { return padr_ratio_; }
- inline float padt_ratio() const { return padt_ratio_; }
- protected:
- std::vector<DetectObject> Postproc(const std::vector<std::pair<float*, uint64_t>>& net_outputs) override;
- private:
-
- float padl_ratio_ = 0, padb_ratio_ = 0, padr_ratio_ = 0, padt_ratio_ = 0;
- };
- }
- #endif
|