如何从列表中自动选择结果

Question

这是我的分段任意模型的输出：

这些是分割结果：

masks = [
    mask['segmentation']
    for mask
    in sorted(sam_result, key=lambda x: x['area'], reverse=True)
]

sv.plot_images_grid(
    images=masks,
    grid_size=(8, int(len(masks) / 8)),
    size=(16, 16)
)

列表中的第二个分段是我想要的。我的目标是用它来消除背景并维护汽车。

但是，我想自动执行此操作，虽然我可以自动化该过程，但需要注意的是我将拥有不同形状的汽车。如何自动分割整个汽车？这该怎么办？

Answer 1

使用显着目标检测，然后将边界框提供给 SAM？