From 82de20e190be7e09e9b8c5a242d3914548c4ebb1 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 11 Aug 2019 15:03:35 +0200 Subject: [PATCH] Added support for Caffe implementation to debug module Signed-off-by: Jim Martens --- src/twomartens/masterthesis/debug.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/twomartens/masterthesis/debug.py b/src/twomartens/masterthesis/debug.py index 64a62d4..f3fd0c7 100644 --- a/src/twomartens/masterthesis/debug.py +++ b/src/twomartens/masterthesis/debug.py @@ -87,6 +87,12 @@ def save_ssd_train_images(images: Union[np.ndarray, Sequence[str]], labels: np.n ymin = instance[4] xmax = instance[5] ymax = instance[6] + elif len(instance) == 6: # predictions using Caffe method + class_id = int(instance[0]) + xmin = instance[2] + ymin = instance[3] + xmax = instance[4] + ymax = instance[5] else: instance = np.copy(instance) class_id = np.argmax(instance[:-12], axis=0)