remove useless heads

Author: upczww

common.hpp

@@ -8,8 +8,8 @@
 #include <opencv2/opencv.hpp>
 #include <dirent.h>
 #include "NvInfer.h"
+#include "NvInferPlugin.h"
 #include "yololayer.h"
-#include "batchedNMSPlugin.h"
 
 #define CHECK(status)                                          \
     do                                                         \

yolov5.cpp

@@ -3,7 +3,6 @@
 #include "cuda_runtime_api.h"
 #include "logging.h"
 #include "common.hpp"
-#include "NvInferPlugin.h"
 
 #define USE_FP16 // comment out this if want to use FP32
 #define DEVICE 0 // GPU id