wehub-resource-sync
26 lines
932 B
Python
26 lines
932 B
Python
from PIL import Image
|
|
import numpy as np
|
|
|
|
def segment_image(image, segmentation_mask):
|
|
image_array = np.array(image)
|
|
segmented_image_array = np.zeros_like(image_array)
|
|
segmented_image_array[segmentation_mask] = image_array[segmentation_mask]
|
|
segmented_image = Image.fromarray(segmented_image_array)
|
|
black_image = Image.new("RGB", image.size, (0, 0, 0))
|
|
transparency_mask = np.zeros_like(segmentation_mask, dtype=np.uint8)
|
|
transparency_mask[segmentation_mask] = 255
|
|
transparency_mask_image = Image.fromarray(transparency_mask, mode='L')
|
|
black_image.paste(segmented_image, mask=transparency_mask_image)
|
|
return black_image
|
|
|
|
|
|
def convert_box_xywh_to_xyxy(box):
|
|
x1 = box[0]
|
|
y1 = box[1]
|
|
x2 = box[0] + box[2]
|
|
y2 = box[1] + box[3]
|
|
return [x1, y1, x2, y2]
|
|
|
|
|
|
def get_indices_of_values_above_threshold(values, threshold):
|
|
return [i for i, v in enumerate(values) if v > threshold] |