support numpy 2.x

This commit is contained in:
chflame163
2025-03-12 13:51:14 +08:00
parent 7b326d13e4
commit 67f664efdc
6 changed files with 7 additions and 9 deletions
+2 -5
View File
@@ -374,13 +374,10 @@ def chop_image(background_image:Image, layer_image:Image, blend_mode:str, opacit
def chop_image_v2(background_image:Image, layer_image:Image, blend_mode:str, opacity:int) -> Image:
backdrop_prepped = np.asfarray(background_image.convert('RGBA'))
source_prepped = np.asfarray(layer_image.convert('RGBA'))
backdrop_prepped = np.asarray(background_image.convert('RGBA'), dtype=float)
source_prepped = np.asarray(layer_image.convert('RGBA'), dtype=float)
blended_np = BLEND_MODES[blend_mode](backdrop_prepped, source_prepped, opacity / 100)
# final_tensor = (torch.from_numpy(blended_np / 255)).unsqueeze(0)
# return tensor2pil(_tensor)
return Image.fromarray(np.uint8(blended_np)).convert('RGB')
def remove_background(image:Image, mask:Image, color:str) -> Image: