DeepLab V3 Plus的高性能Pytorch实现
介绍
此存储库是(重)实现的PyTorch中的语义图像分割,用于在PASCAL VOC数据集上进行语义图像分割。 此回购协议的mIuU高于纸面结果的78.85%,为79.19%。
要求
在运行脚本之前,需要Python(3.6)和Pytorch(0.4.1)。 要安装所需的python软件包(期望PyTorch),请运行
pip install - r requirements . txt
数据集
为了训练和验证网络,此存储库使用增强的PASCAL VOC 2012数据集,其中包含10582张用于训练的图像和1449张用于验证的图像。 要使用数据集,您可以在下载PASCAL VOC培训/验证数据(2GB的tar文件)然后从或下载SegmentationClassAug
训练
在训练之前,您应该克隆此仓库:
git clone gi
2021-12-25 19:10:04
220KB
Python
1