Fix AttributeError: module 'numpy' has no attribute 'float'
This commit is contained in:
1 parent
ac45dd43e4
commit
8612e012ab
1 file changed
+3
-3
@@ -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)
|
# target_image_parsing = postprocess(target_image, source_image, tgt_mask)
|
||||||
|
|
||||||
if use_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:
|
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)
|
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 /= 255
|
||||||
# target_image = 0
|
# 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):
|
for img_mask, target_image in zip(img_mask_list, target_image_list):
|
||||||
img = img_mask * target_image + (1-img_mask) * img
|
img = img_mask * target_image + (1-img_mask) * img
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user