#421 · Primary category: Computer Vision

Multi-Task-Learning-PyTorch

computer-vision eccv2020 multi-task-learning nyud pascal pytorch scene-understanding segmentation

PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).

Project last updated:01/13/22

GitHub Stars

842

Forks

111

Contributors

3

License

Other

Why we included this project

If you work on dense prediction tasks like semantic segmentation, depth estimation, or surface normals, this repo gives you a working set of multi-task learning models to compare. It bundles several architectures and training strategies that share representations across tasks, tested on NYUD and PASCAL, and includes MTI-Net from ECCV 2020, which models cross-task interactions at multiple scales. The config-driven setup makes it easy to swap backbones or datasets once you adjust the path files. Because it accompanies a published survey on multi-task learning for dense prediction, it's a good starting point for reproducing paper results or building your own baseline. Just be ready for some manual setup and a specific PyTorch version; this is research code, not a plug-and-play library.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category