support numpy 2.x
This commit is contained in:
+2
-5
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user