用Pytorch复现常用语义分割网络,备用
Pytorch-SegToolbox
|--backbones 存放主干网络ResNet、AlignedXception等
|--datasets 存放数据集、观察数据工具等
|--data_list 存放数据集的csv文件
|--models 存放语义分割网络 DeepLabV3+(Xception|ResNet101)、UNet(ResNet)、FCN8s等
|--res 存放资源 权重文件、图像、日志等
|--utils 存放工具
|--augment数据增强
|--lossfn损失函数
|--tools日志、计时、评价标准metrics等工具
|--test.py 用来测试
|--train.py 用来训练
由于训练资源(租用GPU)太烧钱,只训练了DeepLabV3+(Xception)和FCN8s。
权重文件,以及以DeepLabV3+(Xception)的测试输出的图片可以在此处下载:
链接:https://pan.baidu.com/s/1U0lx83uKexpfYPaTen3vog
提取码:iyqd
res下的preds只展示了几张效果不错的测试输出,效果不好的,我就不贴了。哈哈……