CenterNet TensorRT works. TensorRT serialization not yet implemented

Signed-oof-by: Davide Sapienza <sapienza.dav@gmail.com>
2019-12-20 11:05:03 +01:00
parent e99b353d8b
commit d889ed385d
9 changed files with 167 additions and 35 deletions
@@ -119,10 +119,10 @@ void Conv2d::inferCUDNN(dnnType* srcData, bool back) {

 Conv2d::Conv2d( Network *net, int out_ch, int kernelH, int kernelW,
                int strideH, int strideW, int paddingH, int paddingW,
-                std::string fname_weights, bool batchnorm, bool deConv) :
+                std::string fname_weights, bool batchnorm, bool deConv, bool final) :
    
    LayerWgs(net, net->getOutputDim().c, out_ch, kernelH, kernelW, 1, 
-             fname_weights, batchnorm) {
+             fname_weights, batchnorm, false, final) {

    this->kernelH = kernelH;
    this->kernelW = kernelW;