Skip to content

Support Multitask in mmseg

Jayden Hsiao requested to merge mmseg_multitask into main

Created by: echonax07

Hello everyone, I am adding all of you as reviewers so that everyone can understand how we will be modifying the code, creating pull requests, and conducting code reviews.

The template for creating PR was already given by the mmseg developer; I just modified parts of it 😄 .

Motivation

To enable multitask learning in segmentation

Modification

Changes have been made in Dataset, model, Metric and Custom hooks.

A detailed description of all the changes is available at https://github.com/Fernando961226/sea-ice-mmseg/blob/mmseg_multitask/docs_ai4arctic/multitask_readme.md

Also, https://github.com/astokholm/AI4ArcticSeaIceChallenge repo has been added as a folder called AI4ArcticSeaIceChallenge.

Use cases

The template config file to be used can be found here: configs/multi_task_ai4arctic/mae_ai4arctic_ds5_pt_80_ft_20.py

Checklist

  1. Pre-commit or other linting tools are used to fix the potential lint issues.
  2. The modification is covered by complete unit tests. If not, please add more unit test to ensure the correctness.
  3. If the modification has potential influence on downstream projects, this PR should be tested with downstream projects, like MMDet or MMDet3D.
  4. The documentation has been modified accordingly, like docstring or example tutorials.

Merge request reports