From 84d6533cb8112f23a34d3de76435a10f4620f4b8 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Mon, 23 Oct 2017 13:43:03 +0000
Subject: [PATCH] Fixed OpenCV usage in the yolo_console_dll.cpp
---
src/parser.c | 14 +++++++++++---
1 files changed, 11 insertions(+), 3 deletions(-)
diff --git a/src/parser.c b/src/parser.c
index 87a6491..ae9e194 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -29,6 +29,7 @@
#include "shortcut_layer.h"
#include "softmax_layer.h"
#include "utils.h"
+#include <stdint.h>
typedef struct{
char *type;
@@ -826,7 +827,7 @@
}
#endif
fprintf(stderr, "Saving weights to %s\n", filename);
- FILE *fp = fopen(filename, "w");
+ FILE *fp = fopen(filename, "wb");
if(!fp) file_error(filename);
int major = 0;
@@ -835,7 +836,7 @@
fwrite(&major, sizeof(int), 1, fp);
fwrite(&minor, sizeof(int), 1, fp);
fwrite(&revision, sizeof(int), 1, fp);
- fwrite(net.seen, sizeof(int), 1, fp);
+ fwrite(net.seen, sizeof(uint64_t), 1, fp);
int i;
for(i = 0; i < net.n && i < cutoff; ++i){
@@ -1023,7 +1024,14 @@
fread(&major, sizeof(int), 1, fp);
fread(&minor, sizeof(int), 1, fp);
fread(&revision, sizeof(int), 1, fp);
- fread(net->seen, sizeof(int), 1, fp);
+ if ((major * 10 + minor) >= 2) {
+ fread(net->seen, sizeof(uint64_t), 1, fp);
+ }
+ else {
+ int iseen = 0;
+ fread(&iseen, sizeof(int), 1, fp);
+ *net->seen = iseen;
+ }
int transpose = (major > 1000) || (minor > 1000);
int i;
--
Gitblit v1.10.0