"""模型骨架：backbone -> neck -> heads。

约束：部署在实时设备上，只使用基于 convolution 的结构（不含 attention / transformer）。
"""

from models import backbones, heads, necks  # noqa: F401  触发注册
from models.build import build_model
from models.defect_net import DefectNet
from models.postprocess import decode_heatmap
from models.registry import BACKBONES, HEADS, MODELS, NECKS

__all__ = [
    "build_model",
    "DefectNet",
    "decode_heatmap",
    "BACKBONES",
    "NECKS",
    "HEADS",
    "MODELS",
]