models/__init__.py

from .backbones import __all__
from .bbox import __all__
from .pqr3d import PQR3D
from .pqr3d_head import PQR3DHead
from .pqr3d_transformer import PQR3DTransformer
from .focal_head import FocalHead, HungarianAssigner2D  # <<< FocalHead

__all__ = [
    'PQR3D', 'PQR3DHead', 'PQR3DTransformer',
    'FocalHead', 'HungarianAssigner2D'
]