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

2023-01-02 15:36:00
aiadmin
原创
2197

这个新蚁合由俄罗斯的步伐员小哥Pavel Yakubovskiy一手打制,包括四种模子架构和30种预操练骨干模子(backbone),官方文档枚举了四条首要特质:

授人以鱼不如授人以渔,放出了model zoo还不算,小哥还知心地供应了何如应用CamVid数据集举行操练的示例。

俄罗斯小哥用这个数据集操练了一个能识别出图片中汽车的新模子。无论这辆车是潜伏正在暗夜之中照旧只要黄豆巨细,都遁不出图像离散模子的手掌心。

修建模子时他采用了传奇的Unet架构,并应用了se_resnext50_32x4d这个预操练骨干模子。

听上去近似很厉害的形态,但全体都确信眼睹为实,不看看模子操练后果何如,心坎不免有一丢丢猜疑。

Pavel Yakubovskiy小哥还正在Colab上放了用Catalyst(PyTorch上的高级框架)操练模子的代码,感兴味的话可能直接正在浏览器上跑一下。

原本Pavel Yakubovskiy小哥早正在旧年就揭橥了一个基于Keras的图像离散model zoo,已经推出就颇受迎接。

但是基于Keras的版本预操练模子要少少少,小哥也没有放出对应的示例,后果何如,就须要我方去搜检啦~

联系我们
联系人: 王先生
电话: 15640228768
微信: 1735252255
地址: 沈阳市铁西区兴华南街58-6号