Add draw method to Yolo3Detection class.

Update yolo demo Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-03-03 17:08:25 +01:00
parent 1d38d88c85
commit c94159a2bf
3 changed files with 33 additions and 23 deletions
@@ -56,7 +56,7 @@ class Yolo3Detection {
         * @return Success of the initialization
         */
        bool init(std::string tensor_path);
-
+        cv::Mat draw(cv::Mat &frame);
        void update(cv::Mat &frame);

        tk::dnn::Yolo* getYoloLayer(int n=0) {