mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Add csresnext50-panet-spp test. Works on CUDNN. Does not work with TensorRT

Browse Source 
Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
This commit is contained in:
Francesco Gatti
2020-03-11 18:37:26 +01:00
parent e6aa73d7bc
commit f6527f51e3
9 changed files with 666 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/tkDNN/Layer.h
+3 -2
View File
@@ -321,12 +321,13 @@ public:
int winH, winW;
int strideH, strideW;
int paddingH, paddingW;
bool test;
tkdnnPoolingMode_t pool_mode;
Pooling(Network *net, int winH, int winW,
int strideH, int strideW,
int paddingH = 0, int paddingW = 0,
tkdnnPoolingMode_t pool_mode = POOLING_MAX, bool final = false);
tkdnnPoolingMode_t pool_mode = POOLING_MAX, bool final = false, bool test=false);
virtual ~Pooling();
virtual layerType_t getLayerType() { return LAYER_POOLING; };
@@ -473,7 +474,7 @@ public:
dnnType *predictions;
static const int MAX_DETECTIONS = 1024;
static const int MAX_DETECTIONS = 4096;
static Yolo::detection *allocateDetections(int nboxes, int classes);
static void mergeDetections(Yolo::detection *dets, int ndets, int classes);
};