| # Copyright (c) OpenMMLab. All rights reserved. | |
| from mmdet.models import SingleStageDetector | |
| from mmdet.registry import MODELS | |
| from mmdet.utils import ConfigType, OptConfigType, OptMultiConfig | |
| class DiffusionDet(SingleStageDetector): | |
| """Implementation of `DiffusionDet <>`_""" | |
| def __init__(self, | |
| backbone: ConfigType, | |
| neck: ConfigType, | |
| bbox_head: ConfigType, | |
| train_cfg: OptConfigType = None, | |
| test_cfg: OptConfigType = None, | |
| data_preprocessor: OptConfigType = None, | |
| init_cfg: OptMultiConfig = None) -> None: | |
| super().__init__( | |
| backbone=backbone, | |
| neck=neck, | |
| bbox_head=bbox_head, | |
| train_cfg=train_cfg, | |
| test_cfg=test_cfg, | |
| data_preprocessor=data_preprocessor, | |
| init_cfg=init_cfg) | |