语义细分的pytorch
该存储库包含一些用于语义细分的模型以及在Pytorch中实施的培训和测试模型的管道
型号
- Vanilla FCN:FCN32,FCN16,FCN8,分别为VGG,Resnet和Densenet(用于语义分割的完全卷积网络)
- U-NET(U-NET:生物医学图像分割的卷积网络)
- segnet(Segnet:用于图像分割的深卷积编码器架构)
- PSPNET(金字塔场景解析网络)
- GCN(大内核很重要)
- DUC,HDC(了解语义分割的卷积)
要求
- Pytorch 0.2.0
- Pytorch的张板。在这里安装
- 其他一些库(在运行代码时找到您错过的内容:p)
准备
- 转到型号目录,并在config.py中设置验证模型的路径
- 转到数据集目录并遵循读取文件
托多
- DeepLab V3
- 炼油
- 更多数据集(例如ADE)