From eef9f8e5bb9fdbe22c6301a56f12e315f9ac64d9 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 23 May 2018 15:35:08 +0000
Subject: [PATCH] Minor fixes
---
src/network.h | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/src/network.h b/src/network.h
index b609546..01a6ab9 100644
--- a/src/network.h
+++ b/src/network.h
@@ -52,6 +52,7 @@
int h, w, c;
int max_crop;
int min_crop;
+ int flip; // horizontal flip 50% probability augmentaiont for classifier training (default = 1)
float angle;
float aspect;
float exposure;
@@ -137,6 +138,7 @@
YOLODLL_API detection *make_network_boxes(network *net, float thresh, int *num);
YOLODLL_API void free_detections(detection *dets, int n);
YOLODLL_API void reset_rnn(network *net);
+YOLODLL_API network *load_network_custom(char *cfg, char *weights, int clear, int batch);
YOLODLL_API network *load_network(char *cfg, char *weights, int clear);
YOLODLL_API float *network_predict_image(network *net, image im);
YOLODLL_API void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, int ngpus, int clear, int dont_show);
--
Gitblit v1.10.0