pytorch-semantic-segmentation

收录理由

这个仓库把语义分割领域几篇经典论文的模型实现收在了一起，FCN、U-Net、SegNet、PSPNet、GCN，还有DUC/HDC，都共用一套训练和测试流程。想对比不同网络结构时，不用在多个项目之间来回切换，直接看这个仓库就能理清各自的思路。代码是那个时期PyTorch的典型写法，结构紧凑，跟着走一遍就能看清每个架构的差异点在哪里。不过它锁定了老版本的PyTorch，而且从2019年起就没有再维护过，所以更适合当作移植到现代代码时的参考，别指望直接拿去跑生产。今天想复现这些模型的话，翻翻这里的原始实现细节，再去看官方版本，会更有底。