14 lines
459 B
C++
14 lines
459 B
C++
#include "batched_nms/trt_batched_nms.hpp"
|
|
#include "nms/trt_nms.hpp"
|
|
#include "roi_align/trt_roi_align.hpp"
|
|
#include "scatternd/trt_scatternd.hpp"
|
|
|
|
REGISTER_TENSORRT_PLUGIN(TRTBatchedNMSPluginDynamicCreator);
|
|
REGISTER_TENSORRT_PLUGIN(NonMaxSuppressionDynamicCreator);
|
|
REGISTER_TENSORRT_PLUGIN(ONNXScatterNDDynamicCreator);
|
|
REGISTER_TENSORRT_PLUGIN(RoIAlignPluginDynamicCreator);
|
|
|
|
extern "C" {
|
|
bool initLibMMCVInferPlugins() { return true; }
|
|
} // extern "C"
|