-
PyTorch中MobileNetv2与DeepLabv3结合的实现:DeepLabv3_MobileNetv2_PyTorch
资源介绍
DeepLabv3_MobileNetv2
这是MobileNet v2网络的PyTorch实施,具有用于语义分割的DeepLab v3结构。
MobileNetv2的骨干来自纸面:
DeepLabv3的段头来自纸面:
如果您对这些块有一些困惑,请参考这些文件,以获取有关诸如Atrous卷积,反向残差,深度卷积或ASPP之类的详细信息。
结果
在训练了150个纪元之后,没有进行任何进一步的调整,测试集上的第一个训练结果如下:
随时更改此仓库中的任何配置或代码:-)
如何使用?
首先,您需要安装此实现的依赖项。 此实现是在Python 3.5下使用以下库编写的:
火炬0.4.0 火炬视觉0.2.1 numpy的1.14.5 OpenCVPython的3.4.1.15 tensorflow 1.8.0(tensorboardX必需) tensorboardX 1.2
使用sudo