GPU Batched NMS

2022-06-19 03:25:50 -03:00
parent 2300e3b44b
commit f621c0f429
24 changed files with 835 additions and 654 deletions
--- a/nvdsinfer_custom_impl_Yolo/layers/dropout_layer.cpp
+++ b/nvdsinfer_custom_impl_Yolo/layers/dropout_layer.cpp
@@ -1,15 +0,0 @@
-/*
- * Created by Marcos Luciano
- * https://www.github.com/marcoslucianops
- */
-
-#include "dropout_layer.h"
-
-nvinfer1::ILayer* dropoutLayer(
-    float probability,
-    nvinfer1::ITensor* input,
-    nvinfer1::INetworkDefinition* network)
-{
-    nvinfer1::ILayer* output;
-    return output;
-}
--- a/nvdsinfer_custom_impl_Yolo/layers/dropout_layer.h
+++ b/nvdsinfer_custom_impl_Yolo/layers/dropout_layer.h
@@ -1,16 +0,0 @@
-/*
- * Created by Marcos Luciano
- * https://www.github.com/marcoslucianops
- */
-
-#ifndef __DROPOUT_LAYER_H__
-#define __DROPOUT_LAYER_H__
-
-#include "NvInfer.h"
-
-nvinfer1::ILayer* dropoutLayer(
-    float probability,
-    nvinfer1::ITensor* input,
-    nvinfer1::INetworkDefinition* network);
-
-#endif