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

interpret

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-22 01:32:59 -07:00
parent 5a52de17eb
commit 5595b8037b
8 changed files with 106 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/pluginsRT/RegionRT.cpp
+1 -4
View File
@@ -4,12 +4,11 @@
class RegionRT : public IPlugin {
public:
RegionRT(int classes, int coords, int num, float thresh) {
RegionRT(int classes, int coords, int num) {
this->classes = classes;
this->coords = coords;
this->num = num;
this->thresh = thresh;
}
~RegionRT(){
@@ -78,7 +77,6 @@ public:
tkDNN::writeBUF(buf, classes);
tkDNN::writeBUF(buf, coords);
tkDNN::writeBUF(buf, num);
tkDNN::writeBUF(buf, thresh);
tkDNN::writeBUF(buf, c);
tkDNN::writeBUF(buf, h);
tkDNN::writeBUF(buf, w);
@@ -86,7 +84,6 @@ public:
int c, h, w;
int classes, coords, num;
float thresh;
int entry_index(int batch, int location, int entry, int batchSize) {
int n = location / (w*h);