mmyolo/projects/assigner_visualization/dense_heads/__init__.py

6 lines
204 B
Python

# Copyright (c) OpenMMLab. All rights reserved.
from .rtmdet_head_assigner import RTMHeadAssigner
from .yolov5_head_assigner import YOLOv5HeadAssigner
__all__ = ['YOLOv5HeadAssigner', 'RTMHeadAssigner']