From 92b25a72006a77073ee99dc69557ce0cde01cc3c Mon Sep 17 00:00:00 2001
From: iovodov <b@ovdv.ru>
Date: Thu, 03 May 2018 09:43:48 +0000
Subject: [PATCH] FIX #736: Detector does not draw rects for images with small resolutions

---
 src/cost_layer.h |    3 ++-
 1 files changed, 2 insertions(+), 1 deletions(-)

diff --git a/src/cost_layer.h b/src/cost_layer.h
index 9ad3124..a692831 100644
--- a/src/cost_layer.h
+++ b/src/cost_layer.h
@@ -1,7 +1,7 @@
 #ifndef COST_LAYER_H
 #define COST_LAYER_H
-#include "params.h"
 #include "layer.h"
+#include "network.h"
 
 typedef layer cost_layer;
 
@@ -10,6 +10,7 @@
 cost_layer make_cost_layer(int batch, int inputs, COST_TYPE type, float scale);
 void forward_cost_layer(const cost_layer l, network_state state);
 void backward_cost_layer(const cost_layer l, network_state state);
+void resize_cost_layer(cost_layer *l, int inputs);
 
 #ifdef GPU
 void forward_cost_layer_gpu(cost_layer l, network_state state);

--
Gitblit v1.10.0