Skip to content
Snippets Groups Projects
Unverified Commit d88f393f authored by Miao Zheng's avatar Miao Zheng Committed by GitHub
Browse files

[Fix] Remove dependcy mmdet when do not use `MaskFormerHead` and `MMDET_Mask2FormerHead` (#2448)

## Motivation

Calling `mmseg.utils.register_all_modules` will import `MaskFormerHead`
and `Mask2FormerHead`, it will crash if mmdet is not installed as `None`
cannot be initialized.

## Modification

- Modify `MMDET_MaskFormerHead=BaseModule` and `MMDET_Mask2FormerHead =
BaseModule` when cannot import from mmdet
parent 115552d5
No related branches found
No related tags found
No related merge requests found
...@@ -4,12 +4,13 @@ from typing import List, Tuple ...@@ -4,12 +4,13 @@ from typing import List, Tuple
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from mmengine.model import BaseModule
try: try:
from mmdet.models.dense_heads import \ from mmdet.models.dense_heads import \
Mask2FormerHead as MMDET_Mask2FormerHead Mask2FormerHead as MMDET_Mask2FormerHead
except ModuleNotFoundError: except ModuleNotFoundError:
MMDET_Mask2FormerHead = None MMDET_Mask2FormerHead = BaseModule
from mmengine.structures import InstanceData from mmengine.structures import InstanceData
from torch import Tensor from torch import Tensor
......
...@@ -4,11 +4,12 @@ from typing import List, Tuple ...@@ -4,11 +4,12 @@ from typing import List, Tuple
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from mmengine.model import BaseModule
try: try:
from mmdet.models.dense_heads import MaskFormerHead as MMDET_MaskFormerHead from mmdet.models.dense_heads import MaskFormerHead as MMDET_MaskFormerHead
except ModuleNotFoundError: except ModuleNotFoundError:
MMDET_MaskFormerHead = None MMDET_MaskFormerHead = BaseModule
from mmengine.structures import InstanceData from mmengine.structures import InstanceData
from torch import Tensor from torch import Tensor
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment