Multi-Task-Learning-PyTorch

收录理由

如果你在做语义分割、深度估计或表面法线这类稠密预测任务，这个仓库能直接给你一套多任务学习模型做对比。它把多种共享表征的架构和训练策略打包在一起，在NYUD和PASCAL上验证过，还收录了ECCV 2020的MTI-Net，专门建模跨任务在多个尺度上的交互。配置驱动的设计让你改改路径文件就能换骨干网络或数据集。由于它配套发表了一篇多任务学习综述，想复现论文结果或搭自己的基线，从这里入手很合适。不过要有点心理准备，环境配置需要手动来，对PyTorch版本也有要求，毕竟是研究代码，不是开箱即用的库。