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/detector.c | 60 +++++++++++++++++++++++++++++++++++++-----------------------
1 files changed, 37 insertions(+), 23 deletions(-)
diff --git a/src/detector.c b/src/detector.c
index 31c44c0..367b3a3 100644
--- a/src/detector.c
+++ b/src/detector.c
@@ -9,6 +9,16 @@
#ifdef OPENCV
#include "opencv2/highgui/highgui_c.h"
+#include "opencv2/core/core_c.h"
+#include "opencv2/core/version.hpp"
+#ifndef CV_VERSION_EPOCH
+#include "opencv2/videoio/videoio_c.h"
+#pragma comment(lib, "opencv_world320.lib")
+#else
+#pragma comment(lib, "opencv_core2413.lib")
+#pragma comment(lib, "opencv_imgproc2413.lib")
+#pragma comment(lib, "opencv_highgui2413.lib")
+#endif
#endif
static int coco_ids[] = {1,2,3,4,5,6,7,8,9,10,11,13,14,15,16,17,18,19,20,21,22,23,24,25,27,28,31,32,33,34,35,36,37,38,39,40,41,42,43,44,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,64,65,67,70,72,73,74,75,76,77,78,79,80,81,82,84,85,86,87,88,89,90};
@@ -78,7 +88,7 @@
int count = 0;
//while(i*imgs < N*120){
while(get_current_batch(net) < net.max_batches){
- if(l.random && count++%10 == 0){
+ if(l.random && count++%10 == 0){
printf("Resizing\n");
int dim = (rand() % 10 + 10) * 32;
if (get_current_batch(net)+100 > net.max_batches) dim = 544;
@@ -137,14 +147,14 @@
i = get_current_batch(net);
printf("%d: %f, %f avg, %f rate, %lf seconds, %d images\n", get_current_batch(net), loss, avg_loss, get_current_rate(net), sec(clock()-time), i*imgs);
- if(i%1000==0 || (i < 1000 && i%100 == 0)){
+ if (i % 1000 == 0 || (i < 1000 && i % 100 == 0)) {
#ifdef GPU
- if(ngpus != 1) sync_nets(nets, ngpus, 0);
+ if (ngpus != 1) sync_nets(nets, ngpus, 0);
#endif
- char buff[256];
- sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i);
- save_weights(net, buff);
- }
+ char buff[256];
+ sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i);
+ save_weights(net, buff);
+ }
free_data(train);
}
#ifdef GPU
@@ -358,7 +368,7 @@
fprintf(stderr, "Total Detection Time: %f Seconds\n", (double)(time(0) - start));
}
-void validate_detector_recall(char *cfgfile, char *weightfile)
+void validate_detector_recall(char *datacfg, char *cfgfile, char *weightfile)
{
network net = parse_network_cfg(cfgfile);
if(weightfile){
@@ -368,7 +378,9 @@
fprintf(stderr, "Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay);
srand(time(0));
- list *plist = get_paths("data/voc.2007.test");
+ list *options = read_data_cfg(datacfg);
+ char *valid_images = option_find_str(options, "valid", "data/train.txt");
+ list *plist = get_paths(valid_images);
char **paths = (char **)list_to_array(plist);
layer l = net.layers[net.n-1];
@@ -382,7 +394,7 @@
int m = plist->size;
int i=0;
- float thresh = .001;
+ float thresh = .2;// .001;
float iou_thresh = .5;
float nms = .4;
@@ -405,6 +417,7 @@
find_replace(labelpath, "JPEGImages", "labels", labelpath);
find_replace(labelpath, ".jpg", ".txt", labelpath);
find_replace(labelpath, ".JPEG", ".txt", labelpath);
+ find_replace(labelpath, ".png", ".txt", labelpath);
int num_labels = 0;
box_label *truth = read_boxes(labelpath, &num_labels);
@@ -413,16 +426,16 @@
++proposals;
}
}
- for (j = 0; j < num_labels; ++j) {
- ++total;
- box t = {truth[j].x, truth[j].y, truth[j].w, truth[j].h};
- float best_iou = 0;
- for(k = 0; k < l.w*l.h*l.n; ++k){
- float iou = box_iou(boxes[k], t);
- if(probs[k][0] > thresh && iou > best_iou){
- best_iou = iou;
- }
- }
+ for (j = 0; j < num_labels; ++j) {
+ ++total;
+ box t = { truth[j].x, truth[j].y, truth[j].w, truth[j].h };
+ float best_iou = 0;
+ for (k = 0; k < l.w*l.h*l.n; ++k) {
+ float iou = box_iou(boxes[k], t);
+ if (probs[k][0] > thresh && iou > best_iou) {
+ best_iou = iou;
+ }
+ }
avg_iou += best_iou;
if(best_iou > iou_thresh){
++correct;
@@ -496,8 +509,9 @@
void run_detector(int argc, char **argv)
{
+ char *out_filename = find_char_arg(argc, argv, "-out_filename", 0);
char *prefix = find_char_arg(argc, argv, "-prefix", 0);
- float thresh = find_float_arg(argc, argv, "-thresh", .25);
+ float thresh = find_float_arg(argc, argv, "-thresh", .24);
int cam_index = find_int_arg(argc, argv, "-c", 0);
int frame_skip = find_int_arg(argc, argv, "-s", 0);
if(argc < 4){
@@ -536,12 +550,12 @@
if(0==strcmp(argv[2], "test")) test_detector(datacfg, cfg, weights, filename, thresh);
else if(0==strcmp(argv[2], "train")) train_detector(datacfg, cfg, weights, gpus, ngpus, clear);
else if(0==strcmp(argv[2], "valid")) validate_detector(datacfg, cfg, weights);
- else if(0==strcmp(argv[2], "recall")) validate_detector_recall(cfg, weights);
+ else if(0==strcmp(argv[2], "recall")) validate_detector_recall(datacfg, cfg, weights);
else if(0==strcmp(argv[2], "demo")) {
list *options = read_data_cfg(datacfg);
int classes = option_find_int(options, "classes", 20);
char *name_list = option_find_str(options, "names", "data/names.list");
char **names = get_labels(name_list);
- demo(cfg, weights, thresh, cam_index, filename, names, classes, frame_skip, prefix);
+ demo(cfg, weights, thresh, cam_index, filename, names, classes, frame_skip, prefix, out_filename);
}
}
--
Gitblit v1.10.0