yoloRT load anchors

include/NetworkRT.h

@@ -1,6 +1,7 @@
 #ifndef NETWORKRT_H
 #define NETWORKRT_H
 
+#include <string.h> // memcpy
 #include "utils.h"
 #include "Network.h"
 #include "Layer.h"
@@ -8,6 +9,40 @@
 
 namespace tk { namespace dnn {
 
+template<typename T> void writeBUF(char*& buffer, const T& val)
+{
+    *reinterpret_cast<T*>(buffer) = val;
+    buffer += sizeof(T);
+}
+
+template<typename T> T readBUF(const char*& buffer)
+{
+    T val = *reinterpret_cast<const T*>(buffer);
+    buffer += sizeof(T);
+    return val;
+}
+
+using namespace nvinfer1;
+#include "pluginsRT/ActivationLeakyRT.h"
+#include "pluginsRT/ReorgRT.h"
+#include "pluginsRT/RegionRT.h"
+//#include "pluginsRT/RouteRT.h"
+#include "pluginsRT/ShortcutRT.h"
+#include "pluginsRT/YoloRT.h"
+#include "pluginsRT/UpsampleRT.h"
+//#include "pluginsRT/Int8Calibrator.h"
+
+class PluginFactory : IPluginFactory
+{
+public:
+    YoloRT *yolos[16];
+    int n_yolos;
+
+	virtual IPlugin* createPlugin(const char* layerName, const void* serialData, size_t serialLength);
+};
+
+
+
 class NetworkRT {
 
 public:
@@ -27,6 +62,8 @@ public:
     dnnType *output;
     cudaStream_t stream;
 
+    PluginFactory *pluginFactory;
+
     NetworkRT(Network *net, const char *name);
     virtual ~NetworkRT();
 
@@ -53,19 +90,5 @@ public:
     bool deserialize(const char *filename);
 };
 
-
-template<typename T> void writeBUF(char*& buffer, const T& val)
-{
-    *reinterpret_cast<T*>(buffer) = val;
-    buffer += sizeof(T);
-}
-
-template<typename T> T readBUF(const char*& buffer)
-{
-    T val = *reinterpret_cast<const T*>(buffer);
-    buffer += sizeof(T);
-    return val;
-}
-
 }}
 #endif //NETWORKRT_H