From 8612e012ab071567106e100007a8e595cbfb63c7 Mon Sep 17 00:00:00 2001 From: Wok Date: Thu, 15 Feb 2024 22:01:46 +0100 Subject: [PATCH] Fix AttributeError: module 'numpy' has no attribute 'float' --- util/reverse2original.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/util/reverse2original.py b/util/reverse2original.py index 83f1bdb..fb7b193 100644 --- a/util/reverse2original.py +++ b/util/reverse2original.py @@ -154,9 +154,9 @@ def reverse2wholeimage(b_align_crop_tenor_list,swaped_imgs, mats, crop_size, ori # target_image_parsing = postprocess(target_image, source_image, tgt_mask) if use_mask: - target_image = np.array(target_image, dtype=np.float) * 255 + target_image = np.array(target_image, dtype=np.float32) * 255 else: - target_image = np.array(target_image, dtype=np.float)[..., ::-1] * 255 + target_image = np.array(target_image, dtype=np.float32)[..., ::-1] * 255 img_mask_list.append(img_mask) @@ -165,7 +165,7 @@ def reverse2wholeimage(b_align_crop_tenor_list,swaped_imgs, mats, crop_size, ori # target_image /= 255 # target_image = 0 - img = np.array(oriimg, dtype=np.float) + img = np.array(oriimg, dtype=np.float32) for img_mask, target_image in zip(img_mask_list, target_image_list): img = img_mask * target_image + (1-img_mask) * img