save layer names in rt file

2019-09-16 19:41:59 +02:00
parent 77f031c0f4
commit ec02c7292f
9 changed files with 50 additions and 6 deletions
@@ -52,6 +52,13 @@ class Yolo3Detection {

        void update(cv::Mat &frame);

+        tk::dnn::Yolo* getYoloLayer(int n=0) {
+            if(n<3)
+                return yolo[n];
+            else 
+                return nullptr;
+        }
+
 };

 }}