Function mask_to_bkgd

datasets/data_utils.py:67–75

(img_path, mask_path)

    return xmin, ymin, xmax, ymax

def mask_to_bkgd(img_path, mask_path):
    img = Image.open(img_path)
    img_array = np.array(img)
    
    mask = Image.open(mask_path).convert("RGB")
    mask_array = np.array(mask)
    
    img_array = np.where(mask_array > 0, img_array, 0)
    return Image.fromarray(img_array)