mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

yoloRT layer

Browse Source
This commit is contained in:
Francesco Gatti
2018-12-22 21:26:50 +01:00
parent 3b60de00f8
commit 34be4cd00f
4 changed files with 120 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/NetworkRT.h
+1
View File
@@ -44,6 +44,7 @@ public:
nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Reorg *l);
nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Region *l);
nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Shortcut *l);
nvinfer1::ILayer* convert_layer(nvinfer1::ITensor *input, Yolo *l);
bool serialize(const char *filename);
bool deserialize(const char *filename);