Add getMemoryUsage function, detection update moved in abstract lass, splitted execution time in pre-inf-post, other minors.

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-01 19:18:43 +02:00
parent e2225d2449
commit 43567dc3ea
14 changed files with 255 additions and 128 deletions
@@ -26,8 +26,7 @@ public:

    bool init(const std::string& tensor_path, const int n_classes=80);
    void preprocess(cv::Mat &frame);
-    void update(cv::Mat &frame);
-    void postprocess(dnnType **rt_out, const int n_out);
+    void postprocess();
 };