fix visualdl bugs

dba4432d · FlyingQianMM · 764455a5 · dba4432d
显示空白变更内容
内联并排

Showing with 3 addition and 3 deletion

paddlex/cv/transforms/visualize.py paddlex/cv/transforms/visualize.py +3 -3

未找到文件。
--- 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]