From 3fc3fd0f1f1924acb7195fe98fe68648e4056a09 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Wed, 28 Mar 2018 23:38:35 +0000
Subject: [PATCH] Fixed training, calculation mAP and anchors for Yolo v3
---
src/data.c | 3 +++
1 files changed, 3 insertions(+), 0 deletions(-)
diff --git a/src/data.c b/src/data.c
index d9cedf5..88d5720 100644
--- a/src/data.c
+++ b/src/data.c
@@ -827,6 +827,9 @@
} else if (a.type == IMAGE_DATA){
*(a.im) = load_image_color(a.path, 0, 0);
*(a.resized) = resize_image(*(a.im), a.w, a.h);
+ }else if (a.type == LETTERBOX_DATA) {
+ *(a.im) = load_image_color(a.path, 0, 0);
+ *(a.resized) = letterbox_image(*(a.im), a.w, a.h);
} else if (a.type == TAG_DATA){
*a.d = load_data_tag(a.paths, a.n, a.m, a.classes, a.min, a.max, a.size, a.angle, a.aspect, a.hue, a.saturation, a.exposure);
}
--
Gitblit v1.10.0