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
@@ -2,6 +2,7 @@
 #define LAYER_H

 #include<iostream>
+#include<vector>
 #include "utils.h"
 #include "Network.h"

@@ -359,6 +360,7 @@ public:
    int classes, num;
    dnnType *mask_h, *mask_d; //anchors
    dnnType *bias_h, *bias_d; //anchors
+    std::vector<std::string> classesNames;

    virtual dnnType* infer(dataDim_t &dim, dnnType* srcData);
    int computeDetections(Yolo::detection *dets, int &ndets, int netw, int neth, float thresh);