Add grouped convolutions in CUDNN and tensorRT.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-02-05 14:32:58 +01:00
parent fe85c26888
commit 4616be0738
5 changed files with 25 additions and 15 deletions
@@ -8,8 +8,13 @@ namespace tk { namespace dnn {

 LayerWgs::LayerWgs(Network *net, int inputs, int outputs, 
                   int kh, int kw, int kl, 
-                   std::string fname_weights, bool batchnorm, bool additional_bias, bool final) : Layer(net, final) {
-
+                   std::string fname_weights, bool batchnorm, bool additional_bias, bool final, bool deConv, int groups) : Layer(net, final) {
+                       
+    if(deConv)
+        inputs = inputs/groups;
+    else
+        outputs = outputs/groups;    
+        
    this->inputs  = inputs;
    this->outputs = outputs;    
    this->weights_path  = std::string(fname_weights);