Function create_long_image

semantic_sam/build_semantic_sam.py:84–92

(masks)

    plt.imshow(image_ori)
    plt.savefig('../vis/input.png')
    def create_long_image(masks):
        ims = []
        for img in masks:
            ims.append(img)
        width, height = ims[0].size
        result = Image.new(ims[0].mode, (width * len(ims), height))
        for i, im in enumerate(ims):
            result.paste(im, box=(i * width, 0))
        return result
    create_long_image(iou_sort_masks).save('../vis/all_results_sort_by_iou.png')
    create_long_image(area_sort_masks).save('../vis/all_results_sort_by_areas.png')