Can now specify input and output names to NetworkRT

2020-12-15 16:39:08 -06:00
parent 714d0685a9
commit f6884f07bf
2 changed files with 7 additions and 7 deletions
@@ -73,7 +73,7 @@ public:

    PluginFactory *pluginFactory;

-    NetworkRT(Network *net, const char *name);
+    NetworkRT(Network *net, const char *name, const char *input_name="data", const char *output_name="out");
    virtual ~NetworkRT();

    int getMaxBatchSize() {