diff --git a/paddlex/cv/transforms/visualize.py b/paddlex/cv/transforms/visualize.py index 2bce491b31edbfd01a490fb75ed9bcd30a060f93..514ac73b2107caade31a5e760b1ec6fa2d4d8076 100644 --- a/paddlex/cv/transforms/visualize.py +++ b/paddlex/cv/transforms/visualize.py @@ -85,6 +85,8 @@ def cls_compose(im, label=None, transforms=None, vdl_writer=None, step=0): im = outputs[0] if len(outputs) == 2: label = outputs[1] + if isinstance(op, pdx.cv.transforms.cls_transforms.Normalize): + continue else: import imgaug.augmenters as iaa if isinstance(op, iaa.Augmenter): @@ -196,7 +198,7 @@ def det_compose(im, im_info=None, label_info=None, transforms=None, vdl_writer=N cname, catid2color[outputs[2]['gt_class'][i][0]-1]) if isinstance(op, pdx.cv.transforms.det_transforms.Normalize): - vdl_im = im + continue else: im = execute_imgaug(op, im) if label_info is not None: @@ -243,6 +245,8 @@ def seg_compose(im, im_info=None, label=None, transforms=None, vdl_writer=None, im_info = outputs[1] if len(outputs) == 3: label = outputs[2] + if isinstance(op, pdx.cv.transforms.seg_transforms.Normalize): + continue else: im = execute_imgaug(op, im) if label is not None: