CenterNet TensorRT serialization works

This commit adds the Deformable layer serialization.

Signed-oof-by: Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
Davide Sapienza
2019-12-23 15:31:13 +01:00
parent f0ea2c027f
commit 5272f1cde6
4 changed files with 188 additions and 24 deletions
+2
View File
@@ -43,6 +43,8 @@ public:
dataDim_t input_dim, output_dim;
dnnType *dstData; //where results will be putted
int id = 0;
bool final; //if the layer is the final one
std::string getLayerName() {