From a7a2e1bb4b0efa55ac2af91358e8c8d2d20076a7 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Fri, 30 Mar 2018 10:19:53 +0000
Subject: [PATCH] Example of anchors generation for Yolo v3
---
src/network.c | 8 +++++++-
1 files changed, 7 insertions(+), 1 deletions(-)
diff --git a/src/network.c b/src/network.c
index 8619158..175c102 100644
--- a/src/network.c
+++ b/src/network.c
@@ -374,10 +374,14 @@
resize_maxpool_layer(&l, w, h);
}else if(l.type == REGION){
resize_region_layer(&l, w, h);
+ }else if (l.type == YOLO) {
+ resize_yolo_layer(&l, w, h);
}else if(l.type == ROUTE){
resize_route_layer(&l, net);
}else if (l.type == SHORTCUT) {
resize_shortcut_layer(&l, w, h);
+ }else if (l.type == UPSAMPLE) {
+ resize_upsample_layer(&l, w, h);
}else if(l.type == REORG){
resize_reorg_layer(&l, w, h);
}else if(l.type == AVGPOOL){
@@ -544,7 +548,9 @@
dets[j].classes = l.classes;
dets[j].bbox = boxes[j];
dets[j].objectness = 1;
- for (i = 0; i < l.classes; ++i) dets[j].prob[i] = probs[j][i];
+ for (i = 0; i < l.classes; ++i) {
+ dets[j].prob[i] = probs[j][i];
+ }
}
free(boxes);
--
Gitblit v1.10.0