From cbc9984a17b3452da4fee397aac912f1e9f7a4c3 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 10 Jun 2015 07:11:41 +0000
Subject: [PATCH] NIPS
---
src/convolutional_layer.c | 9 +++++++++
1 files changed, 9 insertions(+), 0 deletions(-)
diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c
index 67c36c3..e29d995 100644
--- a/src/convolutional_layer.c
+++ b/src/convolutional_layer.c
@@ -227,6 +227,15 @@
return float_to_image(w,h,c,l.filters+i*h*w*c);
}
+void rgbgr_filters(convolutional_layer l)
+{
+ int i;
+ for(i = 0; i < l.n; ++i){
+ image im = get_convolutional_filter(l, i);
+ if (im.c == 3) rgbgr_image(im);
+ }
+}
+
image *get_filters(convolutional_layer l)
{
image *filters = calloc(l.n, sizeof(image));
--
Gitblit v1.10.0