[ADD] Add a python demo.

2020-06-17 11:47:00 +08:00
parent 285c77d6dd
commit 15564efdbb
6 changed files with 342 additions and 2 deletions
@@ -138,7 +138,19 @@ class DetectionNN {
                TKDNN_TSTOP
                if(save_times) *times<<t_ns<<"\n";
            }
-        }      
+        }
+        /**
+         * Method to get results.
+         *
+         */
+        std::vector<std::vector<tk::dnn::box>>& get_batch_detected()
+        {
+            return batchDetected;
+        }
+        std::vector<std::string> get_classesName()
+        {
+            return classesNames;
+        }

        /**
         * Method to draw boundixg boxes and labels on a frame.