From 9ba0beb413d8b443376c36c1701a0dbce1e8d566 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Thu, 17 May 2018 14:02:42 +0000
Subject: [PATCH] Fixed resize of network for low resolution network like 128x128
---
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