From dba4432dfec39a3802cb2a9b89be020e46a70b6c Mon Sep 17 00:00:00 2001
From: FlyingQianMM <245467267@qq.com>
Date: Thu, 13 Aug 2020 15:52:05 +0800
Subject: [PATCH] fix visualdl bugs

---
 paddlex/cv/transforms/visualize.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/paddlex/cv/transforms/visualize.py b/paddlex/cv/transforms/visualize.py
index 19b2384..ef2f451 100644
--- a/paddlex/cv/transforms/visualize.py
+++ b/paddlex/cv/transforms/visualize.py
@@ -191,9 +191,9 @@ def det_compose(im,
                     bboxes[:, 3] = bboxes[:, 3] * h_scale
                 else:
                     bboxes = outputs[2]['gt_bbox']
-                if not isinstance(
-                        op,
-                        pdx.cv.transforms.det_transforms.RandomHorizontalFlip):
+                if not isinstance(op, (
+                        pdx.cv.transforms.det_transforms.RandomHorizontalFlip,
+                        pdx.cv.transforms.det_transforms.Padding)):
                     for i in range(bboxes.shape[0]):
                         bbox = bboxes[i]
                         cname = labels[outputs[2]['gt_class'][i][0] - 1]
-- 
GitLab