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
@@ -4,10 +4,10 @@

 namespace tk { namespace dnn {

-Layer::Layer(Network *net) {
+Layer::Layer(Network *net, bool final) {

    this->net = net;
-
+    this->final = final;
    if(net != nullptr) {
        this->input_dim = net->getOutputDim();
        this->output_dim = input_dim;