mirrors
/
mmyolo
mirror of https://github.com/open-mmlab/mmyolo.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix rtmdet deploy error (#287)

pull/292/head
hanrui1sensetime 2022-11-17 15:53:03 +08:00 committed by GitHub
parent c043181149
commit 5ecaf18eda
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mmyolo/deploy/object_detection.py
View File

@ -16,6 +16,13 @@ class MMYOLO(MMCodebase):
task_registry = MMYOLO_TASK
@classmethod
def register_deploy_modules(cls):
"""register all rewriters for mmcls."""
import mmdeploy.codebase.mmdet.models # noqa: F401
import mmdeploy.codebase.mmdet.ops # noqa: F401
import mmdeploy.codebase.mmdet.structures # noqa: F401
@classmethod
def register_all_modules(cls):
from mmdet.utils.setup_env import \
@ -23,6 +30,8 @@ class MMYOLO(MMCodebase):
from mmyolo.utils.setup_env import \
register_all_modules as register_all_modules_mmyolo
cls.register_deploy_modules()
register_all_modules_mmyolo(True)
register_all_modules_mmdet(False)