fix retinaface

retinaface/CMakeLists.txt

@@ -25,6 +25,7 @@ endif()
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -Wall -Ofast -Wfatal-errors -D_MWAITXINTRIN_H_INCLUDED")
 
 cuda_add_library(decodeplugin SHARED ${PROJECT_SOURCE_DIR}/decode.cu)
+target_link_libraries(decodeplugin nvinfer cudart)
 
 find_package(OpenCV)
 include_directories(OpenCV_INCLUDE_DIRS)

retinaface/decode.h

@@ -113,6 +113,7 @@ namespace nvinfer1
             static PluginFieldCollection mFC;
             static std::vector<PluginField> mPluginAttributes;
     };
+    REGISTER_TENSORRT_PLUGIN(DecodePluginCreator);
 };
 
 #endif 

retinaface/retina_r50.cpp

@@ -34,7 +34,6 @@ const char* OUTPUT_BLOB_NAME = "prob";
 
 using namespace nvinfer1;
 static Logger gLogger;
-REGISTER_TENSORRT_PLUGIN(DecodePluginCreator);
 
 cv::Mat preprocess_img(cv::Mat& img) {
     int w, h, x, y;