mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source 
## 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
This commit is contained in:
Miao Zheng 2022-12-31 01:02:58 +08:00 committed by GitHub
parent 115552d5ea
commit d88f393ffa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmseg/models/decode_heads/mask2former_head.py
View File

@ -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

3
mmseg/models/decode_heads/maskformer_head.py
View File

@ -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