Centernet: fix pooling problem, add centrnet demo

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com> Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-01-21 12:50:18 +01:00
parent c68d6f318e
commit 7f239efdc0
12 changed files with 229 additions and 935 deletions
@@ -93,6 +93,9 @@ class CenternetDetection {
        // draw
        std::vector<std::string> coco_class_name;

+        // keep track of inference times (ms)
+        std::vector<double> stats;
+        
        CenternetDetection() {}

        virtual ~CenternetDetection() {}