MNN/express/module/NMSModule.hpp

29 lines
586 B
C++
Raw Normal View History

2022-01-04 10:50:40 +08:00
//
// NMSModule.hpp
// MNN
//
// Created by MNN on b'2020/09/10'.
// Copyright © 2018, Alibaba Group Holding Limited
//
#ifndef NMSModule_hpp
#define NMSModule_hpp
#include <MNN/expr/Module.hpp>
namespace MNN {
namespace Express {
class NMSModule : public Module {
public:
virtual ~ NMSModule() {
// Do nothing
}
virtual std::vector<Express::VARP> onForward(const std::vector<Express::VARP>& inputs) override;
2022-02-18 11:30:27 +08:00
MNN_PUBLIC static NMSModule* create(const Op* op);
2022-01-04 10:50:40 +08:00
private:
NMSModule(){}
Module* clone(CloneContext* ctx) const override;
};
}
}
#endif