From 2afa376bb37b379f27954f74859fbfa63402ea46 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 14 Aug 2015 18:45:11 +0000
Subject: [PATCH] single shot yolo training, separate file
---
src/imagenet.c | 4 ++++
1 files changed, 4 insertions(+), 0 deletions(-)
diff --git a/src/imagenet.c b/src/imagenet.c
index 499a8d3..aeb7e69 100644
--- a/src/imagenet.c
+++ b/src/imagenet.c
@@ -2,6 +2,10 @@
#include "utils.h"
#include "parser.h"
+#ifdef OPENCV
+#include "opencv2/highgui/highgui_c.h"
+#endif
+
void train_imagenet(char *cfgfile, char *weightfile)
{
data_seed = time(0);
--
Gitblit v1.10.0