MMSegmentation語(yǔ)義分割工具箱

MMSegmentation 是一個(gè)基于 PyTorch 的語(yǔ)義分割開(kāi)源工具箱。它是 OpenMMLab 項(xiàng)目的一部分，目前支持 PyTorch 1.5 以上的版本。

主要特性

• 統(tǒng)一的基準(zhǔn)平臺(tái)

  我們將各種各樣的語(yǔ)義分割算法集成到了一個(gè)統(tǒng)一的工具箱，進(jìn)行基準(zhǔn)測(cè)試。

• 模塊化設(shè)計(jì)

  MMSegmentation 將分割框架解耦成不同的模塊組件，通過(guò)組合不同的模塊組件，用戶可以便捷地構(gòu)建自定義的分割模型。

• 豐富的即插即用的算法和模型

  MMSegmentation 支持了眾多主流的和最新的檢測(cè)算法，例如 PSPNet，DeepLabV3，PSANet，DeepLabV3+ 等.

• 速度快

  訓(xùn)練速度比其他語(yǔ)義分割代碼庫(kù)更快或者相當(dāng)。

基準(zhǔn)測(cè)試和模型庫(kù)

測(cè)試結(jié)果和模型可以在模型庫(kù)中找到。

已支持的骨干網(wǎng)絡(luò)：

•  ResNet (CVPR'2016)
•  ResNeXt (CVPR'2017)
•  HRNet (CVPR'2019)
•  ResNeSt (ArXiv'2020)
•  MobileNetV2 (CVPR'2018)
•  MobileNetV3 (ICCV'2019)

已支持的算法：

•  FCN (CVPR'2015/TPAMI'2017)
•  ERFNet (T-ITS'2017)
•  UNet (MICCAI'2016/Nat. Methods'2019)
•  PSPNet (CVPR'2017)
•  DeepLabV3 (ArXiv'2017)
•  BiSeNetV1 (ECCV'2018)
•  PSANet (ECCV'2018)
•  DeepLabV3+ (CVPR'2018)
•  UPerNet (ECCV'2018)
•  ICNet (ECCV'2018)
•  NonLocal Net (CVPR'2018)
•  EncNet (CVPR'2018)
•  Semantic FPN (CVPR'2019)
•  DANet (CVPR'2019)
•  APCNet (CVPR'2019)
•  EMANet (ICCV'2019)
•  CCNet (ICCV'2019)
•  DMNet (ICCV'2019)
•  ANN (ICCV'2019)
•  GCNet (ICCVW'2019/TPAMI'2020)
•  FastFCN (ArXiv'2019)
•  Fast-SCNN (ArXiv'2019)
•  ISANet (ArXiv'2019/IJCV'2021)
•  OCRNet (ECCV'2020)
•  DNLNet (ECCV'2020)
•  PointRend (CVPR'2020)
•  CGNet (TIP'2020)
•  BiSeNetV2 (IJCV'2021)
•  SETR (CVPR'2021)
•  DPT (ArXiv'2021)
•  SegFormer (ArXiv'2021)

已支持的數(shù)據(jù)集：

•  Cityscapes
•  PASCAL VOC
•  ADE20K
•  Pascal Context
•  COCO-Stuff 10k
•  COCO-Stuff 164k
•  CHASE_DB1
•  DRIVE
•  HRF
•  STARE
•  Dark Zurich
•  Nighttime Driving
•  LoveDA
•   Vision Transformer (ICLR'2021)
•  Swin Transformer (ArXiv'2021)