Segmentation Models PyTorch:高效建構語義分割模型的開源利器

Segmentation Models PyTorch(SMP)是一個基於 PyTorch 的語義分割深度學習庫,透過統一的高層 API 極大簡化了複雜視覺任務的開發。它整合了 12 種主流編碼器-解碼器架構,支援超過 800 個預訓練卷積和 Transformer 基座模型,涵蓋從 ResNet 到 SegFormer 的廣泛選擇。SMP 內建了 Dice、Jaccard 等流行指標與損失函數,相容 ONNX 匯出和 Torch Compile 優化。適用於醫學影像、自動駕駛、工業缺陷檢測等高精度像素級分類場景。

Sources