上传者: 42110070
|
上传时间: 2021-11-03 14:39:51
|
文件大小: 1.52MB
|
文件类型: -
PyTorch用于语义分割
该存储库包含一些用于语义分割的模型以及在PyTorch中实现的训练和测试模型的管道
楷模
Vanilla FCN:分别为VGG,ResNet和DenseNet版本的FCN32,FCN16,FCN8( )
U-Net( )
SegNet( )
PSPNet()
GCN()
DUC,HDC()
需求
PyTorch 0.2.0
PyTorch的TensorBoard。 安装
其他一些库(在运行代码时查找丢失的内容:-P)
制备
转到models目录并在config.py中设置预训练模型的路径
转到数据集目录并按照自述文件进行操作
去做
DeepLab v3