From a6c51e3b758aee7fd3a6f1d37daa8dcad4891e52 Mon Sep 17 00:00:00 2001 From: AlexeyAB <alexeyab84@gmail.com> Date: Thu, 08 Mar 2018 22:42:58 +0000 Subject: [PATCH] Fixes for small objects. And fixes of densenet201_yolo.cfg, resnet50_yolo.cfg, resnet152_yolo.cfg --- src/parser.h | 3 +++ 1 files changed, 3 insertions(+), 0 deletions(-) diff --git a/src/parser.h b/src/parser.h index b16cc03..b760116 100644 --- a/src/parser.h +++ b/src/parser.h @@ -3,8 +3,11 @@ #include "network.h" network parse_network_cfg(char *filename); +network parse_network_cfg_custom(char *filename, int batch); void save_network(network net, char *filename); void save_weights(network net, char *filename); +void save_weights_upto(network net, char *filename, int cutoff); +void save_weights_double(network net, char *filename); void load_weights(network *net, char *filename); void load_weights_upto(network *net, char *filename, int cutoff); -- Gitblit v1.10.0