PyTorch-Encoding

收录理由

这个工具包把多篇经典论文中的编码模块集中在一起，包括 Context Encoding、Deep TEN，以及 ResNeSt 的分裂注意力主干网络。如果你正在复现或搭建语义分割模型，可以直接把这些模块拿过来用，省去自己重新实现的功夫。它同时提供了同步批归一化，以及 ImageNet 分类和 ADE20K、Pascal Context 分割的预训练权重，这样就不必从零训练或手动对接检查点。文档里还写了训练脚本和配方，适合想扩展现有网络结构、而不是学习新框架的人参考。