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

serialize fix

Browse Source
This commit is contained in:
Francesco Gatti
2020-06-14 13:01:45 +02:00
parent 567dc0f75d
commit cbfc8ea4f2
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/NetworkRT.cpp
+1 -1
View File
@@ -595,7 +595,7 @@ ILayer* NetworkRT::convert_layer(ITensor *input, DeformConv2d *l) {
bool NetworkRT::serialize(const char *filename) {
std::ofstream p(filename);
std::ofstream p(filename, std::ios::binary);
if (!p) {
FatalError("could not open plan output file");
return false;
tests/darknet/yolo3.cpp
+7 -7
View File
@@ -5,19 +5,19 @@
#include "DarknetParser.h"
int main() {
std::string bin_path = "yolo3";
std::string bin_path = "yolov3-sppx";
std::vector<std::string> input_bins = {
bin_path + "/layers/input.bin"
};
std::vector<std::string> output_bins = {
bin_path + "/debug/layer82_out.bin",
bin_path + "/debug/layer94_out.bin",
bin_path + "/debug/layer106_out.bin"
bin_path + "/debug/layer89_out.bin",
bin_path + "/debug/layer101_out.bin",
bin_path + "/debug/layer113_out.bin"
};
std::string wgs_path = bin_path + "/layers";
std::string cfg_path = "../tests/darknet/cfg/yolo3.cfg";
std::string name_path = "../tests/darknet/names/coco.names";
downloadWeightsifDoNotExist(input_bins[0], bin_path, "https://cloud.hipert.unimore.it/s/jPXmHyptpLoNdNR/download");
std::string cfg_path = "../tests/darknet/cfg/yolov3-sppx.cfg";
std::string name_path = "../tests/darknet/names/coco4.names";
//downloadWeightsifDoNotExist(input_bins[0], bin_path, "https://cloud.hipert.unimore.it/s/jPXmHyptpLoNdNR/download");
// parse darknet network
tk::dnn::Network *net = tk::dnn::darknetParser(cfg_path, wgs_path, name_path);