csresnext50-panet-spp works with TensorRT

Signed-off-by: Francesco Gatti <gattifrancesco@hotmail.it>
2020-03-12 18:24:41 +01:00
parent f6527f51e3
commit 13b4dffd71
8 changed files with 1225 additions and 102 deletions
@@ -321,7 +321,7 @@ public:
    int winH, winW;
    int strideH, strideW;
    int paddingH, paddingW;
-    bool test;
+    bool maxpoolfixedsize;
    tkdnnPoolingMode_t pool_mode;

    Pooling(Network *net, int winH, int winW, 
@@ -474,7 +474,7 @@ public:

    dnnType *predictions;

-    static const int MAX_DETECTIONS = 4096;
+    static const int MAX_DETECTIONS = 1024;
    static Yolo::detection *allocateDetections(int nboxes, int classes);
    static void             mergeDetections(Yolo::detection *dets, int ndets, int classes);
 };