
DeepLabv3_MobileNetv2_PyTorch: PyTorch下的MobileNetv2与DeepLabv3结合实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
本项目在PyTorch框架下实现了基于MobileNetv2骨干网络的DeepLabv3模型,适用于多种场景的语义分割任务。
DeepLabv3_MobileNetv2是基于MobileNet v2网络的PyTorch实现,并集成了用于语义分割的DeepLab v3结构。
该实施使用了以下论文中的骨干部分:
- MobileNetv2
以及以下论文中的段头部分:
- DeepLabv3
如果您对反向残差、深度卷积或ASPP等概念感到困惑,可以参考相关文件获取更多信息。
在训练150个纪元后,并未进行任何额外调整的情况下,在测试集上得到了如下初步结果:
您可以随时更改此仓库中的配置和代码。
首先,请安装该实现所需的依赖项。这个版本是在Python 3.5环境下使用以下库编写的:
- 火炬 (Torch) 版本0.4.0
- 火炬视觉 (Torchvision) 版本0.2.1
- numpy 版本1.14.5
- OpenCV Python版本3.4.1.15
- tensorflow 1.8.0(用于tensorboardX)
- tensorboardX 1.2
全部评论 (0)
还没有任何评论哟~


