From 27a5109542f5973e627cd3118327104aade23a88 Mon Sep 17 00:00:00 2001 From: ShusenTang Date: Tue, 29 Oct 2019 18:39:16 +0800 Subject: [PATCH] fix bug in MultiBoxPrior --- code/d2lzh_pytorch/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/code/d2lzh_pytorch/utils.py b/code/d2lzh_pytorch/utils.py index 0171e42..0e1a4b3 100644 --- a/code/d2lzh_pytorch/utils.py +++ b/code/d2lzh_pytorch/utils.py @@ -777,7 +777,7 @@ def show_bboxes(axes, bboxes, labels=None, colors=None): colors = _make_list(colors, ['b', 'g', 'r', 'm', 'c']) for i, bbox in enumerate(bboxes): color = colors[i % len(colors)] - rect = d2l.bbox_to_rect(bbox.detach().cpu().numpy(), color) + rect = bbox_to_rect(bbox.detach().cpu().numpy(), color) axes.add_patch(rect) if labels and len(labels) > i: text_color = 'k' if color == 'w' else 'w' -- GitLab