一大波PyTorch图像分割模型来袭，俄罗斯程序员出品新model zoo

鱼羊 发自 凹非寺
量子位 报道 | 公众号 QbitAI

一个新的图像分割model zoo来啦！

一大波基于PyTorch的图像分割模型整理好了就等你来用~

这个新集合由俄罗斯的程序员小哥Pavel Yakubovskiy一手打造，包含四种模型架构和30种预训练骨干模型（backbone），官方文档列举了四条主要特点：

• 高级API（两行代码构建神经网络）

• 4种模型架构：Unet，Linknet，FPN，PSPNet

• 每个架构有30种可用的编码器

• 所有编码器都具有预先训练的权重，可实现更快、更好的收敛

授人以鱼不如授人以渔，放出了model zoo还不算，小哥还贴心地提供了如何使用CamVid数据集进行训练的示例。

效果怎么样

CamVid数据集是计算机视觉领域常用的一个数据集，通常用来进行街景分割。

俄罗斯小哥用这个数据集训练了一个能识别出图片中汽车的新模型。无论这辆车是隐藏在暗夜之中还是只有黄豆大小，都逃不出图像分割模型的手掌心。