lightning-sam

收录理由

原版SAM在部分图像上表现不佳，作者以文档中的文字分割为例说明了这一局限。这个库允许你在自己的COCO格式数据集上对SAM进行微调，以边界框作为提示，并基于Lightning Fabric实现。README坦诚地说明这是一个实验性项目，旨在验证框提示微调能否提升掩码质量。它仅支持框提示，包含训练和验证循环，更适合作为参考实现而非生产级流水线。在TrashCan和WGISD数据集上报告的IoU提升，能让你直观感受到微调带来的收益，从而判断在自己的领域投入这项工作的价值。