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

Modify darknetFields_t

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-05-30 18:10:14 +02:00
parent 26e6ba52a5
commit 15e4ab4436
1 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/tkDNN/DarknetParser.h
+13 -10
View File
@@ -9,16 +9,25 @@ namespace tk { namespace dnn {
int height = 0;
int channels = 0;
int batch_normalize=0;
int groups = 0;
int filters=0;
int size=0;
int stride=0;
int pad=0;
int size_x=0;
int size_y=0;
int stride_x=0;
int stride_y=0;
int padding_x = 0;
int padding_y = 0;
int n_mask = 0;
int classes = 0;
int num = 0;
float scale_xy = 0;
std::vector<int> layers;
std::string activation = "";
};
std::ostream& operator<<(std::ostream& os, const darknetFields_t& f){
os << f.width << " " << f.height << " " << f.channels << " " << f.batch_normalize<< " " << f.filters<< " " << f.size<< " " << f.stride << " " << f.pad << " " << f.activation;
os << f.width << " " << f.height << " " << f.channels << " " << f.batch_normalize<< " " << f.filters << " " << " " << f.activation;
return os;
}
@@ -59,12 +68,6 @@ namespace tk { namespace dnn {
fields.batch_normalize = std::stoi(value);
else if (name == "filters")
fields.filters = std::stoi(value);
else if (name == "size")
fields.size = std::stoi(value);
else if (name == "stride")
fields.stride = std::stoi(value);
else if (name == "pad")
fields.pad = std::stoi(value);
else if (name == "activation")
fields.activation = value;